Home
last modified time | relevance | path

Searched refs:bboxDeltasShape (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DGenerateProposals.cpp68 uint32_t numClasses = getSizeOfDimension(bboxDeltasShape, 1) / roiLength; in bboxTransformFloat32()
117 std::vector<float> delta_float32(getNumberOfElements(bboxDeltasShape)); in bboxTransformFloat16()
138 convertQuantToFloat32(bboxDeltasData, bboxDeltasShape.scale, bboxDeltasShape.offset, in bboxTransformQuant()
160 convertQuantToFloat32<int8_t>(bboxDeltasData, bboxDeltasShape.scale, bboxDeltasShape.offset, in bboxTransformQuant()
232 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare() local
238 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 2); in prepare()
250 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numRois); in prepare()
1005 Shape tempBBoxDeltasShape = bboxDeltasShape; in generateProposalsNhwcFloat32Compute()
1183 convertQuantToFloat32<T_8QInput>(bboxDeltasData, bboxDeltasShape.scale, bboxDeltasShape.offset, in generateProposalsQuant()
1292 Shape bboxDeltasShape = context->getInputShape(kDeltaTensor); in prepare() local
[all …]