| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 268 NN_RET_CHECK(context->setOutputShape(kOutputTensor, outputShape)); in prepare()
578 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
582 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
586 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
590 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in boxWithNmsLimitWriteOutput()
786 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
796 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
800 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
804 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, outputBatchSplitShape)); in prepare()
1092 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsFloat32()
[all …]
|
| D | TopK_V2.cpp | 112 return context->setOutputShape(kOutputValuesTensor, outputValuesShape) && in prepare()
113 context->setOutputShape(kOutputIndicesTensor, outputIndicesShape); in prepare()
|
| D | Elementwise.cpp | 138 return context->setOutputShape(kOutputTensor, output); in prepare()
146 return context->setOutputShape(kOutputTensor, output); in prepareFloor()
|
| D | BidirectionalSequenceRNN.cpp | 444 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
451 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
458 NN_RET_CHECK(context->setOutputShape(kFwOutputHiddenStateTensor - delta, in prepare()
460 NN_RET_CHECK(context->setOutputShape(kBwOutputHiddenStateTensor - delta, in prepare()
|
| D | Rank.cpp | 55 return context->setOutputShape(kOutputScalar, output); in prepare()
|
| D | LogicalNot.cpp | 62 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | UnidirectionalSequenceRNN.cpp | 194 NN_RET_CHECK(context->setOutputShape(kStateOutputTensor, outputStateShape)); in prepare()
197 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Neg.cpp | 69 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Elu.cpp | 75 return context->setOutputShape(kOutputTensor, inputShape); in prepare()
|
| D | Fill.cpp | 96 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | LogicalAndOr.cpp | 79 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Quantize.cpp | 93 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Squeeze.cpp | 126 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 385 NN_RET_CHECK(context->setOutputShape(kOutputStateOutTensor, outputStateOutTensor)); in prepare()
391 NN_RET_CHECK(context->setOutputShape(kCellStateOutTensor, cellStateOutTensor)); in prepare()
394 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | LogSoftmax.cpp | 96 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
|
| D | Select.cpp | 106 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | ChannelShuffle.cpp | 92 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| D | Gather.cpp | 101 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Transpose.cpp | 133 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Dequantize.cpp | 112 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | InstanceNormalization.cpp | 128 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| D | HeatmapMaxKeypoint.cpp | 284 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
295 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()
|
| D | PRelu.cpp | 125 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Slice.cpp | 134 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciDriver.cpp | 91 return context->setOutputShape(kOutputTensor, output); in prepare()
|