Searched refs:imageInfoDataShape (Results 1 – 1 of 1) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 69 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32()
119 std::vector<float> imageInfo_float32(getNumberOfElements(imageInfoDataShape)); in bboxTransformFloat16()
140 std::vector<float> imageInfo_float32(getNumberOfElements(imageInfoDataShape)); in bboxTransformQuant()
141 convertQuantToFloat32(imageInfoData, imageInfoDataShape.scale, imageInfoDataShape.offset, in bboxTransformQuant()
163 convertQuantToFloat32(imageInfoData, imageInfoDataShape.scale, imageInfoDataShape.offset, in bboxTransformQuant()
1294 Shape imageInfoDataShape = context->getInputShape(kImageInfoTensor); in prepare() local
1302 NN_RET_CHECK_EQ(getNumberOfDimensions(imageInfoDataShape), 2); in prepare()
1314 NN_RET_CHECK_EQ(getSizeOfDimension(imageInfoDataShape, 0), numBatches); in prepare()
1315 NN_RET_CHECK_EQ(getSizeOfDimension(imageInfoDataShape, 1), 2); in prepare()
1321 NN_RET_CHECK_EQ(imageInfoDataShape.scale, 0.125f); in prepare()
[all …]
|