Searched refs:imageInfoShape (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 233 Shape imageInfoShape = context->getInputShape(kImageInfoTensor); in prepare() local
239 NN_RET_CHECK_EQ(getNumberOfDimensions(imageInfoShape), 2); in prepare()
245 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare()
252 NN_RET_CHECK_EQ(getSizeOfDimension(imageInfoShape, 1), 2); in prepare()
257 NN_RET_CHECK_EQ(imageInfoShape.scale, 0.125f); in prepare()
258 NN_RET_CHECK_EQ(imageInfoShape.offset, 0); in prepare()
955 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsNhwcFloat32Compute() argument
966 uint32_t imageInfoLength = getSizeOfDimension(imageInfoShape, 1); in generateProposalsNhwcFloat32Compute()
1002 Shape tempImageInfoShape = imageInfoShape; in generateProposalsNhwcFloat32Compute()
1057 const float* imageInfoData, const Shape& imageInfoShape, in generateProposalsFloat32Compute() argument
[all …]
|