Searched refs:outputClassShape (Results 1 – 1 of 1) sorted by relevance
765 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare() local803 outputClassShape.type = OperandType::TENSOR_INT32; in prepare()804 outputClassShape.dimensions = {0}; in prepare()805 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()1625 Shape outputClassShape = context->getOutputShape(kOutputClassTensor); in prepare() local1682 outputClassShape.type = OperandType::TENSOR_INT32; in prepare()1683 outputClassShape.dimensions = {numBatches, numOutDetections}; in prepare()1684 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()