| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | GenerateProposals.cpp | 237 NN_RET_CHECK(context->setOutputShape(kOutputTensor, outputShape)); in prepare()
531 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in boxWithNmsLimitWriteOutput()
535 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, roiOutShape)); in boxWithNmsLimitWriteOutput()
539 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, classesOutShape)); in boxWithNmsLimitWriteOutput()
543 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, batchesOutShape)); in boxWithNmsLimitWriteOutput()
698 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScoreShape)); in prepare()
708 NN_RET_CHECK(context->setOutputShape(kOutputRoiTensor, outputRoiShape)); in prepare()
712 NN_RET_CHECK(context->setOutputShape(kOutputClassTensor, outputClassShape)); in prepare()
716 NN_RET_CHECK(context->setOutputShape(kOutputBatchesTensor, outputBatchSplitShape)); in prepare()
986 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, scoresOutShape)); in generateProposalsFloat32()
[all …]
|
| D | TopK_V2.cpp | 83 return context->setOutputShape(kOutputValuesTensor, outputValuesShape) && in prepare()
84 context->setOutputShape(kOutputIndicesTensor, outputIndicesShape); in prepare()
|
| D | BidirectionalSequenceRNN.cpp | 384 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
391 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
398 NN_RET_CHECK(context->setOutputShape(kFwOutputHiddenStateTensor - delta, in prepare()
400 NN_RET_CHECK(context->setOutputShape(kBwOutputHiddenStateTensor - delta, in prepare()
|
| D | Rank.cpp | 31 return context->setOutputShape(kOutputScalar, output); in prepare()
|
| D | UnidirectionalSequenceRNN.cpp | 159 NN_RET_CHECK(context->setOutputShape(kStateOutputTensor, outputStateShape)); in prepare()
162 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Elementwise.cpp | 153 return context->setOutputShape(kOutputTensor, output); in prepare()
161 return context->setOutputShape(kOutputTensor, output); in prepareFloor()
|
| D | LogicalNot.cpp | 43 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Fill.cpp | 52 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | Neg.cpp | 47 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Elu.cpp | 51 return context->setOutputShape(kOutputTensor, inputShape); in prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 279 NN_RET_CHECK(context->setOutputShape(kOutputStateOutTensor, outputStateOutTensor)); in prepare()
285 NN_RET_CHECK(context->setOutputShape(kCellStateOutTensor, cellStateOutTensor)); in prepare()
288 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | Reverse.cpp | 58 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | Squeeze.cpp | 88 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | LogicalAndOr.cpp | 62 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | LogSoftmax.cpp | 66 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
|
| D | Select.cpp | 81 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Quantize.cpp | 67 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | ChannelShuffle.cpp | 59 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| D | HeatmapMaxKeypoint.cpp | 247 NN_RET_CHECK(context->setOutputShape(kOutputScoreTensor, outputScore)); in prepare()
258 NN_RET_CHECK(context->setOutputShape(kOutputKeypointTensor, outputKeypoint)); in prepare()
|
| D | Pack.cpp | 79 return context->setOutputShape(kOutputTensor, outputShape); in prepare()
|
| D | Dequantize.cpp | 79 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | InstanceNormalization.cpp | 99 return context->setOutputShape(kOutputTensor, input); in prepare()
|
| D | Transpose.cpp | 109 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | Gather.cpp | 69 return context->setOutputShape(kOutputTensor, output); in prepare()
|
| D | QLSTM.cpp | 245 return context->setOutputShape(kOutputStateOutTensor, outputShape) && in prepare()
246 context->setOutputShape(kCellStateOutTensor, cellStateOutShape) && in prepare()
247 context->setOutputShape(kOutputTensor, outputShape); in prepare()
|