Searched refs:outputTensorShape (Results 1 – 3 of 3) sorted by relevance
57 const Shape outputTensorShape = context->getOutputShape(kOutputTensor); in validate() local60 NN_RET_CHECK_EQ(inputTensorShape.scale, outputTensorShape.scale) in validate()62 << " does not match output tensor scale " << outputTensorShape.scale; in validate()63 NN_RET_CHECK_EQ(inputTensorShape.offset, outputTensorShape.offset) in validate()65 << " does not match output tensor offset " << outputTensorShape.offset; in validate()77 if (hasKnownRank(outputTensorShape)) { in validate()79 auto outputTensorRank = getNumberOfDimensions(outputTensorShape); in validate()
54 const Shape outputTensorShape = context->getOutputShape(kOutputTensor); in validate() local57 NN_RET_CHECK_EQ(inputTensorShape.scale, outputTensorShape.scale) in validate()59 << " does not match output tensor scale " << outputTensorShape.scale; in validate()60 NN_RET_CHECK_EQ(inputTensorShape.offset, outputTensorShape.offset) in validate()62 << " does not match output tensor offset " << outputTensorShape.offset; in validate()65 auto outputTensorRank = getNumberOfDimensions(outputTensorShape); in validate()
223 Shape outputTensorShape = context->getOutputShape(kOutputTensor); in prepare() local224 outputTensorShape.dimensions = in prepare()226 return context->setOutputShape(kOutputTensor, outputTensorShape); in prepare()