| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Activation.cpp | 404 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
409 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
414 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
419 context->getOutputBuffer<int8_t>(kOutputTensor), in executeRelu()
433 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu1()
438 context->getOutputBuffer<float>(kOutputTensor), in executeRelu1()
443 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu1()
448 context->getOutputBuffer<int8_t>(kOutputTensor), in executeRelu1()
462 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu6()
467 context->getOutputBuffer<float>(kOutputTensor), in executeRelu6()
[all …]
|
| D | Dequantize.cpp | 94 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
96 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
102 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
104 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
110 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
112 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
118 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
121 context->getOutputBuffer<float>(kOutputTensor)); in execute()
|
| D | Elementwise.cpp | 78 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
82 context->getOutputBuffer<float>(kOutputTensor)); in execute()
96 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeAbs()
100 context->getOutputBuffer<float>(kOutputTensor)); in executeAbs()
105 context->getOutputBuffer<int32_t>(kOutputTensor)); in executeAbs()
118 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeRsqrt()
122 context->getOutputBuffer<float>(kOutputTensor)); in executeRsqrt()
131 context->getOutputBuffer<uint8_t>(kOutputTensor)); in executeRsqrt()
141 context->getOutputBuffer<int8_t>(kOutputTensor)); in executeRsqrt()
|
| D | Broadcast.cpp | 461 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
469 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
477 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
485 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
493 context->getOutputBuffer<int32_t>(kOutputTensor), in executeAdd()
511 context->getOutputBuffer<_Float16>(kOutputTensor), in executeMul()
519 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
527 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
535 context->getOutputBuffer<int8_t>(kOutputTensor), in executeMul()
543 context->getOutputBuffer<int32_t>(kOutputTensor), in executeMul()
[all …]
|
| D | GenerateProposals.cpp | 255 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
267 context->getOutputBuffer<float>(kOutputTensor), in execute()
280 context->getOutputBuffer<uint16_t>(kOutputTensor), in execute()
291 context->getOutputBuffer<uint16_t>(kOutputTensor), in execute()
549 T_Score* scoresOutPtr = context->getOutputBuffer<T_Score>(kOutputScoreTensor); in boxWithNmsLimitWriteOutput()
550 T_Roi* roiOutPtr = context->getOutputBuffer<T_Roi>(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
551 int32_t* classesOutPtr = context->getOutputBuffer<int32_t>(kOutputClassTensor); in boxWithNmsLimitWriteOutput()
552 int32_t* batchesOutPtr = context->getOutputBuffer<int32_t>(kOutputBatchesTensor); in boxWithNmsLimitWriteOutput()
739 context->getOutputBuffer<_Float16>(kOutputScoreTensor), in execute()
741 context->getOutputBuffer<_Float16>(kOutputRoiTensor), in execute()
[all …]
|
| D | Gather.cpp | 81 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
87 context->getOutputBuffer<float>(kOutputTensor)); in execute()
93 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
99 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
105 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
|
| D | Quantize.cpp | 79 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
83 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
89 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
93 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | ChannelShuffle.cpp | 70 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
74 context->getOutputBuffer<float>(kOutputTensor)); in execute()
78 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
82 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
|
| D | Neg.cpp | 55 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
59 context->getOutputBuffer<float>(kOutputTensor)); in execute()
63 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
|
| D | HeatmapMaxKeypoint.cpp | 270 auto outputScoreData = context->getOutputBuffer<_Float16>(kOutputScoreTensor); in execute()
272 auto outputKeypointData = context->getOutputBuffer<_Float16>(kOutputKeypointTensor); in execute()
293 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()
295 context->getOutputBuffer<float>(kOutputKeypointTensor), in execute()
305 context->getOutputBuffer<uint8_t>(kOutputScoreTensor), in execute()
307 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
316 context->getOutputBuffer<int8_t>(kOutputScoreTensor), in execute()
318 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
|
| D | Slice.cpp | 121 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
130 context->getOutputBuffer<float>(kOutputTensor), in execute()
139 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
148 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
157 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | Comparisons.cpp | 69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
78 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
87 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
96 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
105 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
114 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterTyped()
|
| D | Transpose.cpp | 122 context->getOutputBuffer<float>(kOutputTensor), in execute()
129 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
136 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
143 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | PRelu.cpp | 120 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
131 context->getOutputBuffer<float>(kOutputTensor), in execute()
138 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
146 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | FullyConnected.cpp | 203 context->getOutputBuffer<float>(kOutputTensor), in execute()
213 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
223 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
233 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | Elu.cpp | 62 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
68 context->getOutputBuffer<float>(kOutputTensor), in execute()
|
| D | BatchMatmul.cpp | 238 context->getOutputBuffer<float>(kOutputTensor), in execute()
247 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
256 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
265 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | UnidirectionalSequenceLSTM.cpp | 307 outputStateOut = context->getOutputBuffer<float>(kOutputStateOutTensor); in execute()
308 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()
349 cellStateOut, context->getOutputBuffer<float>(kOutputTensor), in execute()
359 outputStateOut = context->getOutputBuffer<_Float16>(kOutputStateOutTensor); in execute()
360 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
402 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
|
| D | BidirectionalSequenceRNN.cpp | 133 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped()
139 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()
218 fwOutputHiddenState = context->getOutputBuffer<T>(kFwOutputHiddenStateTensor - delta); in executeTyped()
219 bwOutputHiddenState = context->getOutputBuffer<T>(kBwOutputHiddenStateTensor - delta); in executeTyped()
277 context->getOutputBuffer<T>(kFwOutputTensor)); in executeTyped()
280 context->getOutputBuffer<T>(kBwOutputTensor)); in executeTyped()
|
| D | ResizeImageOps.cpp | 233 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
238 halfPixelCenters, context->getOutputBuffer<float>(kOutputTensor), in execute()
243 halfPixelCenters, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
248 halfPixelCenters, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | Softmax.cpp | 254 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
260 context->getOutputBuffer<float>(kOutputTensor), in execute()
266 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
272 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|
| D | LogicalAndOr.cpp | 69 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
77 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
|
| D | UnidirectionalSequenceRNN.cpp | 64 T* output = context->getOutputBuffer<T>(kOutputTensor); in executeTyped()
105 context->getOutputBuffer<T>(kOutputTensor)); in executeTyped()
110 T* stateOutput = context->getOutputBuffer<T>(kStateOutputTensor); in executeTyped()
|
| D | LogSoftmax.cpp | 77 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
82 context->getOutputBuffer<float>(kOutputTensor)); in execute()
|
| D | L2Normalization.cpp | 235 context->getOutputBuffer<float>(kOutputTensor), in execute()
240 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
245 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
250 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
|