Searched defs: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()
145 const Shape& batchSplitShape, float heightStride, float widthStride, in roiPooling()
162 const Shape& batchSplitShape, float heightStride, in roiPooling()
177 const Shape& batchSplitShape, float heightStride, in roiPooling()
233 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
|
| D | RoiAlign.cpp | 59 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlignNhwc()
184 const int32_t* batchSplitData, const Shape& batchSplitShape, in roiAlignQuantNhwc()
317 const Shape& batchSplitShape, float heightStride, float widthStride, in roiAlign()
388 Shape batchSplitShape = context->getInputShape(kBatchSplitTensor); in prepare() local
|