Home
last modified time | relevance | path

Searched defs:roiShape (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DRoiPooling.cpp43 const Shape& roiShape, const int32_t* batchSplitData, in roiPoolingNhwc()
132 const Shape& roiShape, const int32_t* batchSplitData, in roiPooling()
148 const uint16_t* roiData, const Shape& roiShape, in roiPooling()
163 const uint16_t* roiData, const Shape& roiShape, in roiPooling()
181 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
DRoiAlign.cpp48 const Shape& roiShape, const int32_t* batchSplitData, in roiAlignNhwc()
173 const uint16_t* roiData, const Shape& roiShape, in roiAlignQuantNhwc()
306 const Shape& roiShape, const int32_t* batchSplitData, in roiAlign()
335 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
DGenerateProposals.cpp63 inline bool bboxTransformFloat32(const float* roiData, const Shape& roiShape, in bboxTransformFloat32()
113 inline bool bboxTransformFloat16(const _Float16* roiData, const Shape& roiShape, in bboxTransformFloat16()
133 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant()
155 inline bool bboxTransformQuant(const uint16_t* roiData, const Shape& roiShape, in bboxTransformQuant()
199 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local
571 const Shape& roiShape, const int32_t* batchesData, in boxWithNmsLimitFloat32()
594 const _Float16* roiData, const Shape& roiShape, in boxWithNmsLimitFloat16()
618 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant()
641 const uint16_t* roiData, const Shape& roiShape, in boxWithNmsLimitQuant()
668 Shape roiShape = context->getInputShape(kRoiTensor); in prepare() local