Searched refs:setOutputShape (Results 1 – 25 of 36) sorted by relevance
12
| /frameworks/ml/nn/common/operations/ |
| D | GenerateProposals.cpp | 229 NN_RET_CHECK(context->setOutputShape(kOutputTensor, outputShape)); in prepare()
524 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
528 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
532 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
536 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in boxWithNmsLimitWriteOutput()
703 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
709 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
713 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
717 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, outputBatchSplitShape)); in prepare()
983 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsFloat32()
[all …]
|
| D | LogicalNot.cpp | 59 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Neg.cpp | 67 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Quantize.cpp | 84 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | LogicalAndOr.cpp | 77 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Elementwise.cpp | 80 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Select.cpp | 101 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | ChannelShuffle.cpp | 81 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| D | LogSoftmax.cpp | 94 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
|
| D | HeatmapMaxKeypoint.cpp | 248 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
255 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()
|
| D | Dequantize.cpp | 103 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Gather.cpp | 93 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Transpose.cpp | 120 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | PRelu.cpp | 114 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | InstanceNormalization.cpp | 115 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| D | BidirectionalSequenceRNN.cpp | 365 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
372 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
|
| D | Concatenation.cpp | 154 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Slice.cpp | 122 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | UnidirectionalSequenceRNN.cpp | 165 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | FullyConnected.cpp | 223 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Reduce.cpp | 134 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | ResizeImageOps.cpp | 189 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | L2Normalization.cpp | 187 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Comparisons.cpp | 142 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciDriver.cpp | 88 return context->setOutputShape(kOutputTensor, output); in prepare()
|
12