Searched refs:bboxDeltasShape (Results 1 – 1 of 1) sorted by relevance
/aosp12/packages/modules/NeuralNetworks/common/operations/ |
H A D | GenerateProposals.cpp | 68 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 …]
|