Searched refs:batchSplitShape (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RoiPooling.cpp | 133 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling() argument
140 batchSplitData, batchSplitShape, heightStride, widthStride, in roiPooling()
150 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
156 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
165 const Shape& batchSplitShape, float heightStride, in roiPooling() argument
171 batchSplitShape, heightStride, widthStride, useNchw, outputData, in roiPooling()
182 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
192 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
|
| D | RoiAlign.cpp | 307 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign() argument
318 batchSplitShape, heightStride, widthStride, heightSamplingRatio, widthSamplingRatio, in roiAlign()
322 batchSplitData, batchSplitShape, heightStride, widthStride, in roiAlign()
336 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
351 NN_RET_CHECK_EQ(getSizeOfDimension(batchSplitShape, 0), numRois); in prepare()
|