| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | GenerateProposals.cpp | 269 NN_RET_CHECK(context->setOutputShape(kOutputTensor, outputShape)); in prepare()
581 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
585 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
589 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
801 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
1099 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsFloat32()
1102 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsFloat32()
1154 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsFloat16()
1206 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in generateProposalsQuant()
1336 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
[all …]
|
| H A D | TopK_V2.cpp | 109 return context->setOutputShape(kOutputValuesTensor, outputValuesShape) && in prepare()
110 context->setOutputShape(kOutputIndicesTensor, outputIndicesShape); in prepare()
|
| H A D | BidirectionalSequenceRNN.cpp | 443 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
450 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
457 NN_RET_CHECK(context->setOutputShape(kFwOutputHiddenStateTensor - delta, in prepare()
459 NN_RET_CHECK(context->setOutputShape(kBwOutputHiddenStateTensor - delta, in prepare()
|
| H A D | Elementwise.cpp | 132 return context->setOutputShape(kOutputTensor, output); in prepare()
140 return context->setOutputShape(kOutputTensor, output); in prepareFloor()
|
| H A D | LogicalNot.cpp | 59 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Rank.cpp | 54 return context->setOutputShape(kOutputScalar, output); in prepare()
|
| H A D | UnidirectionalSequenceRNN.cpp | 194 NN_RET_CHECK(context->setOutputShape(kStateOutputTensor, outputStateShape)); in prepare()
197 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Neg.cpp | 66 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Elu.cpp | 74 return context->setOutputShape(kOutputTensor, inputShape); in prepare()
|
| H A D | Fill.cpp | 93 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| H A D | UnidirectionalSequenceLSTM.cpp | 388 NN_RET_CHECK(context->setOutputShape(kOutputStateOutTensor, outputStateOutTensor)); in prepare()
394 NN_RET_CHECK(context->setOutputShape(kCellStateOutTensor, cellStateOutTensor)); in prepare()
397 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| H A D | LogicalAndOr.cpp | 79 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Quantize.cpp | 90 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Select.cpp | 103 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Squeeze.cpp | 124 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| H A D | LogSoftmax.cpp | 94 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
|
| H A D | ChannelShuffle.cpp | 89 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| H A D | HeatmapMaxKeypoint.cpp | 286 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
297 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()
|
| H A D | Dequantize.cpp | 111 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Gather.cpp | 98 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | InstanceNormalization.cpp | 130 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| H A D | PRelu.cpp | 127 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | Transpose.cpp | 137 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| H A D | QLSTM.cpp | 361 return context->setOutputShape(kOutputStateOutTensor, outputShape) && in prepare()
362 context->setOutputShape(kCellStateOutTensor, cellStateOutShape) && in prepare()
363 context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| /aosp12/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/ |
| H A D | FibonacciDriver.cpp | 90 return context->setOutputShape(kOutputTensor, output); in prepare()
|