Home
last modified time | relevance | path

Searched refs:getOutputBuffer (Results 1 – 25 of 73) sorted by relevance

123

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DActivation.cpp450 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu()
455 context->getOutputBuffer<float>(kOutputTensor), in executeRelu()
460 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu()
479 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu1()
484 context->getOutputBuffer<float>(kOutputTensor), in executeRelu1()
489 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeRelu1()
508 context->getOutputBuffer<_Float16>(kOutputTensor), in executeRelu6()
513 context->getOutputBuffer<float>(kOutputTensor), in executeRelu6()
571 context->getOutputBuffer<float>(kOutputTensor), in executeTanh()
576 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeTanh()
[all …]
H A DDequantize.cpp126 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
128 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
134 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
136 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
142 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
144 return compute(inputBuffer, inputShape, context->getOutputBuffer<float>(kOutputTensor)); in execute()
150 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
153 context->getOutputBuffer<float>(kOutputTensor)); in execute()
H A DBroadcast.cpp505 context->getOutputBuffer<_Float16>(kOutputTensor), in executeAdd()
513 context->getOutputBuffer<float>(kOutputTensor), in executeAdd()
521 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeAdd()
529 context->getOutputBuffer<int8_t>(kOutputTensor), in executeAdd()
563 context->getOutputBuffer<float>(kOutputTensor), in executeMul()
571 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeMul()
579 context->getOutputBuffer<int8_t>(kOutputTensor), in executeMul()
613 context->getOutputBuffer<float>(kOutputTensor), in executeSub()
621 context->getOutputBuffer<uint8_t>(kOutputTensor), in executeSub()
629 context->getOutputBuffer<int8_t>(kOutputTensor), in executeSub()
[all …]
H A DGather.cpp110 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
116 context->getOutputBuffer<float>(kOutputTensor)); in execute()
122 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
128 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
134 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
H A DQuantize.cpp102 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
106 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
112 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
116 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DGenerateProposals.cpp600 T_Roi* roiOutPtr = context->getOutputBuffer<T_Roi>(kOutputRoiTensor); in boxWithNmsLimitWriteOutput()
832 context->getOutputBuffer<_Float16>(kOutputScoreTensor), in execute()
834 context->getOutputBuffer<_Float16>(kOutputRoiTensor), in execute()
836 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
838 context->getOutputBuffer<int32_t>(kOutputBatchesTensor), in execute()
1112 float* roiOutData = context->getOutputBuffer<float>(kOutputRoiTensor); in generateProposalsFloat32()
1716 context->getOutputBuffer<_Float16>(kOutputRoiTensor), in execute()
1718 context->getOutputBuffer<int32_t>(kOutputClassTensor), in execute()
1742 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()
1744 context->getOutputBuffer<float>(kOutputRoiTensor), in execute()
[all …]
H A DElementwise.cpp52 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
56 context->getOutputBuffer<float>(kOutputTensor)); in execute()
70 context->getOutputBuffer<_Float16>(kOutputTensor)); in executeAbs()
74 context->getOutputBuffer<float>(kOutputTensor)); in executeAbs()
79 context->getOutputBuffer<int32_t>(kOutputTensor)); in executeAbs()
H A DChannelShuffle.cpp100 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
104 context->getOutputBuffer<float>(kOutputTensor)); in execute()
108 context->getOutputBuffer<uint8_t>(kOutputTensor)); in execute()
112 context->getOutputBuffer<int8_t>(kOutputTensor)); in execute()
H A DNeg.cpp74 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
78 context->getOutputBuffer<float>(kOutputTensor)); in execute()
82 context->getOutputBuffer<int32_t>(kOutputTensor)); in execute()
H A DSlice.cpp153 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
162 context->getOutputBuffer<float>(kOutputTensor), in execute()
171 context->getOutputBuffer<int32_t>(kOutputTensor), in execute()
180 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
189 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DHeatmapMaxKeypoint.cpp309 auto outputScoreData = context->getOutputBuffer<_Float16>(kOutputScoreTensor); in execute()
311 auto outputKeypointData = context->getOutputBuffer<_Float16>(kOutputKeypointTensor); in execute()
332 context->getOutputBuffer<float>(kOutputScoreTensor), in execute()
334 context->getOutputBuffer<float>(kOutputKeypointTensor), in execute()
344 context->getOutputBuffer<uint8_t>(kOutputScoreTensor), in execute()
346 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
355 context->getOutputBuffer<int8_t>(kOutputScoreTensor), in execute()
357 context->getOutputBuffer<uint16_t>(kOutputKeypointTensor), in execute()
H A DComparisons.cpp75 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessTyped()
84 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeLessEqualTyped()
93 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeEqualTyped()
102 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeNotEqualTyped()
111 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterEqualTyped()
120 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeGreaterTyped()
H A DPRelu.cpp141 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
152 context->getOutputBuffer<float>(kOutputTensor), in execute()
159 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
167 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DTranspose.cpp150 context->getOutputBuffer<float>(kOutputTensor), in execute()
157 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
164 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
171 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DElu.cpp85 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
91 context->getOutputBuffer<float>(kOutputTensor), in execute()
H A DBidirectionalSequenceRNN.cpp162 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped()
168 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()
247 fwOutputHiddenState = context->getOutputBuffer<T>(kFwOutputHiddenStateTensor - delta); in executeTyped()
248 bwOutputHiddenState = context->getOutputBuffer<T>(kBwOutputHiddenStateTensor - delta); in executeTyped()
306 context->getOutputBuffer<T>(kFwOutputTensor)); in executeTyped()
309 context->getOutputBuffer<T>(kBwOutputTensor)); in executeTyped()
H A DUnidirectionalSequenceLSTM.cpp416 outputStateOut = context->getOutputBuffer<float>(kOutputStateOutTensor); in execute()
417 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()
458 cellStateOut, context->getOutputBuffer<float>(kOutputTensor), in execute()
468 outputStateOut = context->getOutputBuffer<_Float16>(kOutputStateOutTensor); in execute()
469 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
511 outputStateOut, cellStateOut, context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
H A DLogicalAndOr.cpp86 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeAnd()
94 context->getInputShape(kInputTensor2), context->getOutputBuffer<bool8>(kOutputTensor), in executeOr()
H A DFullyConnected.cpp316 context->getOutputBuffer<float>(kOutputTensor), in execute()
326 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
336 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
346 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DL2Normalization.cpp271 context->getOutputBuffer<float>(kOutputTensor), in execute()
276 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
281 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
286 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DSoftmax.cpp295 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
301 context->getOutputBuffer<float>(kOutputTensor), in execute()
307 context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
313 context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DUnidirectionalSequenceRNN.cpp76 T* output = context->getOutputBuffer<T>(kOutputTensor); in executeTyped()
117 context->getOutputBuffer<T>(kOutputTensor)); in executeTyped()
122 T* stateOutput = context->getOutputBuffer<T>(kStateOutputTensor); in executeTyped()
H A DResizeImageOps.cpp294 context->getOutputBuffer<_Float16>(kOutputTensor), in execute()
299 halfPixelCenters, context->getOutputBuffer<float>(kOutputTensor), in execute()
304 halfPixelCenters, context->getOutputBuffer<uint8_t>(kOutputTensor), in execute()
309 halfPixelCenters, context->getOutputBuffer<int8_t>(kOutputTensor), in execute()
H A DLogSoftmax.cpp105 context->getOutputBuffer<_Float16>(kOutputTensor)); in execute()
110 context->getOutputBuffer<float>(kOutputTensor)); in execute()
H A DTopK_V2.cpp70 context->getOutputBuffer<T>(kOutputValuesTensor), in executeTyped()
71 context->getOutputBuffer<int32_t>(kOutputIndicesTensor)); in executeTyped()

123