Searched defs:batchSplitShape (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | RoiPooling.cpp | 57 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPoolingNhwc()
146 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling()
234 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
|
| H A D | RoiAlign.cpp | 61 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlignNhwc()
186 const int32_t* batchSplitData, const Shape& batchSplitShape, in roiAlignQuantNhwc()
319 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign()
391 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
|