Searched refs:batchSplitShape (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RoiPooling.cpp | 56 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPoolingNhwc() argument
145 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling() argument
152 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
162 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
168 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
177 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
183 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
233 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
243 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
|
| D | RoiAlign.cpp | 59 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlignNhwc() argument
184 const int32_t* batchSplitData, const Shape& batchSplitShape, in roiAlignQuantNhwc() argument
317 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign() argument
328 batchSplitShape, heightStride, widthStride, heightSamplingRatio, widthSamplingRatio, in roiAlign()
332 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
388 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
403 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
|