Searched refs:batchSplitData (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RoiPooling.cpp | 55 const Shape& roiShape, const int32_t* batchSplitData, in roiPoolingNhwc() argument
77 uint32_t batchId = batchSplitData[roiIndex]; in roiPoolingNhwc()
144 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling() argument
152 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
161 const int32_t* batchSplitData, in roiPooling() argument
167 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
176 const int32_t* batchSplitData, in roiPooling() argument
182 NN_RET_CHECK(roiPooling(inputData, inputShape, roi_float32.data(), roiShape, batchSplitData, in roiPooling()
|
| D | RoiAlign.cpp | 58 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc() argument
81 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignNhwc()
184 const int32_t* batchSplitData, const Shape& batchSplitShape, in roiAlignQuantNhwc() argument
208 uint32_t batchId = static_cast<uint32_t>(batchSplitData[roiIndex]); in roiAlignQuantNhwc()
316 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign() argument
327 input.getNhwcBuffer(), input.getNhwcShape(), roiData, roiShape, batchSplitData, in roiAlign()
332 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
|