Searched refs: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() argument
65 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc()
132 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling() argument
140 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
149 const int32_t* batchSplitData, in roiPooling() argument
155 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
164 const int32_t* batchSplitData, in roiPooling() argument
170 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
|
| D | RoiAlign.cpp | 48 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc() argument
71 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc()
174 const int32_t* batchSplitData, const Shape& /*batchSplitShape*/, in roiAlignQuantNhwc() argument
198 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc()
306 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign() argument
317 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
322 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
|