Searched defs:batchSplitData (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RoiPooling.cpp | 43 const Shape& roiShape, const int32_t* batchSplitData, in roiPoolingNhwc()
132 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling()
149 const int32_t* batchSplitData, in roiPooling()
164 const int32_t* batchSplitData, in roiPooling()
|
| D | RoiAlign.cpp | 48 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc()
174 const int32_t* batchSplitData, const Shape& /*batchSplitShape*/, in roiAlignQuantNhwc()
306 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign()
|