Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DGenerateProposals.cpp1626 Shape outputDetectionShape = context->getOutputShape(kOutputDetectionTensor); in prepare() local
1686 outputDetectionShape.type = OperandType::TENSOR_INT32; in prepare()
1687 outputDetectionShape.dimensions = {numBatches}; in prepare()
1688 NN_RET_CHECK(context->setOutputShape(kOutputDetectionTensor, outputDetectionShape)); in prepare()