Searched refs:batchSplitData (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | RoiPooling.cpp | 56 const Shape& roiShape, const int32_t* batchSplitData, in roiPoolingNhwc() argument
78 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc()
145 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling() argument
153 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
162 const int32_t* batchSplitData,
168 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData,
177 const int32_t* batchSplitData,
183 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData,
|
| H A D | RoiAlign.cpp | 60 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc() argument
83 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc()
186 const int32_t* batchSplitData, const Shape& batchSplitShape, in roiAlignQuantNhwc() argument
210 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc()
318 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign() argument
329 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
334 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
|