Searched defs: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()
150 const Shape& batchSplitShape, float heightStride, in roiPooling()
165 const Shape& batchSplitShape, float heightStride, in roiPooling()
182 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
|
| D | RoiAlign.cpp | 307 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign()
336 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
|