Searched defs: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()
145 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling()
|
| H A D | RoiAlign.cpp | 60 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc()
186 const int32_t* batchSplitData, const Shape& batchSplitShape, in roiAlignQuantNhwc()
318 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign()
|