Home
last modified time | relevance | path

Searched refs:getInputBuffer (Results 1 – 25 of 74) sorted by relevance

123

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp95 return context->getInputBuffer(tensor) != nullptr; in hasTensor()
446 context->getInputBuffer<float>(kInputGateBiasTensor), in execute()
447 context->getInputBuffer<float>(kForgetGateBiasTensor), in execute()
448 context->getInputBuffer<float>(kCellGateBiasTensor), in execute()
449 context->getInputBuffer<float>(kOutputGateBiasTensor), in execute()
451 context->getInputBuffer<float>(kProjectionBiasTensor), in execute()
452 context->getInputBuffer<float>(kOutputStateInTensor), in execute()
453 context->getInputBuffer<float>(kCellStateInTensor), in execute()
479 context->getInputBuffer<_Float16>(kInputTensor), in execute()
501 context->getInputBuffer<_Float16>(kCellGateBiasTensor), in execute()
[all …]
H A DSlice.cpp123 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()
124 const int32_t* sizeData = context->getInputBuffer<int32_t>(kSizeTensor); in prepare()
149 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
151 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
156 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
158 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
160 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
169 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
178 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
183 return evalGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
H A DBroadcast.cpp508 return addFloat32(context->getInputBuffer<float>(kInputTensor1), in executeAdd()
510 context->getInputBuffer<float>(kInputTensor2), in executeAdd()
524 return addQuant8(context->getInputBuffer<int8_t>(kInputTensor1), in executeAdd()
526 context->getInputBuffer<int8_t>(kInputTensor2), in executeAdd()
558 return mulFloat32(context->getInputBuffer<float>(kInputTensor1), in executeMul()
560 context->getInputBuffer<float>(kInputTensor2), in executeMul()
574 return mulQuant8(context->getInputBuffer<int8_t>(kInputTensor1), in executeMul()
576 context->getInputBuffer<int8_t>(kInputTensor2), in executeMul()
608 return subFloat32(context->getInputBuffer<float>(kInputTensor1), in executeSub()
610 context->getInputBuffer<float>(kInputTensor2), in executeSub()
[all …]
H A DGather.cpp106 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
108 context->getInputBuffer<int32_t>(kInputIndices), in execute()
112 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
114 context->getInputBuffer<int32_t>(kInputIndices), in execute()
118 return eval(context->getInputBuffer<int32_t>(kInputTensor), in execute()
120 context->getInputBuffer<int32_t>(kInputIndices), in execute()
124 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
126 context->getInputBuffer<int32_t>(kInputIndices), in execute()
130 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
132 context->getInputBuffer<int32_t>(kInputIndices), in execute()
H A DActivation.cpp448 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
453 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
458 return reluQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu()
477 return relu1Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu1()
482 return relu1Float(context->getInputBuffer<float>(kInputTensor), in executeRelu1()
487 return relu1Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu1()
506 return relu6Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu6()
511 return relu6Float(context->getInputBuffer<float>(kInputTensor), in executeRelu6()
569 return tanhFloat32(context->getInputBuffer<float>(kInputTensor), in executeTanh()
574 return tanhQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeTanh()
[all …]
H A DTranspose.cpp120 const int32_t* permData = context->getInputBuffer<int32_t>(kPermTensor); in prepare()
146 return transposeGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
148 context->getInputBuffer<int32_t>(kPermTensor), in execute()
153 return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
155 context->getInputBuffer<int32_t>(kPermTensor), in execute()
160 return transposeGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
162 context->getInputBuffer<int32_t>(kPermTensor), in execute()
167 return transposeGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
169 context->getInputBuffer<int32_t>(kPermTensor), in execute()
H A DComparisons.cpp73 std::less<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessTyped()
74 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
82 std::less_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessEqualTyped()
83 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
91 std::equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeEqualTyped()
92 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
100 std::not_equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeNotEqualTyped()
101 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
109 std::greater_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeGreaterEqualTyped()
110 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
[all …]
H A DPRelu.cpp137 context->getInputBuffer<_Float16>(kInputTensor), in execute()
139 context->getInputBuffer<_Float16>(kAlphaTensor), in execute()
148 context->getInputBuffer<float>(kInputTensor), in execute()
150 context->getInputBuffer<float>(kAlphaTensor), in execute()
155 return evalQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
157 context->getInputBuffer<uint8_t>(kAlphaTensor), in execute()
163 return evalQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
165 context->getInputBuffer<int8_t>(kAlphaTensor), in execute()
H A DFullyConnected.cpp309 return fullyConnectedFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
311 context->getInputBuffer<float>(kWeightsTensor), in execute()
313 context->getInputBuffer<float>(kBiasTensor), in execute()
319 return fullyConnectedFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
323 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
329 return fullyConnectedQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
331 context->getInputBuffer<uint8_t>(kWeightsTensor), in execute()
333 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
339 return fullyConnectedQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
341 context->getInputBuffer<int8_t>(kWeightsTensor), in execute()
[all …]
H A DQLSTM.cpp102 return context->getInputBuffer(tensor) != nullptr; in hasTensor()
407 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputTensor)); in execute()
413 context->getInputBuffer(kRecurrentToInputWeightsTensor)); in execute()
417 reinterpret_cast<const int16_t*>(context->getInputBuffer(kInputLayerNormTensor)); in execute()
419 reinterpret_cast<const int32_t*>(context->getInputBuffer(kInputGateBiasTensor)); in execute()
424 context->getInputBuffer(kRecurrentToForgetWeightsTensor)); in execute()
437 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellLayerNormTensor)); in execute()
439 reinterpret_cast<const int32_t*>(context->getInputBuffer(kCellGateBiasTensor)); in execute()
444 context->getInputBuffer(kRecurrentToOutputWeightsTensor)); in execute()
458 reinterpret_cast<const int8_t*>(context->getInputBuffer(kPrevOutputTensor)); in execute()
[all …]
H A DRoiPooling.cpp280 return roiPooling(context->getInputBuffer<_Float16>(kInputTensor), in execute()
282 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
284 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
292 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
294 context->getInputBuffer<float>(kRoiTensor), in execute()
296 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
304 return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
306 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
308 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
316 return roiPooling(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
H A DTransposeConv2D.cpp80 const int32_t* outputShapeData = context->getInputBuffer<int32_t>(3); in initialize()
553 return transposeConv(context->getInputBuffer<float>(kInputTensor), in execute()
555 context->getInputBuffer<float>(kFilterTensor), in execute()
557 context->getInputBuffer<float>(kBiasTensor), in execute()
566 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
574 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
576 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
581 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
601 context->getInputBuffer<int8_t>(kInputTensor), in execute()
603 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
[all …]
H A DBidirectionalSequenceRNN.cpp122 const T* input = context->getInputBuffer<T>(kInputTensor); in executeTyped()
125 const T* fwWeights = context->getInputBuffer<T>(kFwWeightsTensor); in executeTyped()
129 const T* fwBias = context->getInputBuffer<T>(kFwBiasTensor); in executeTyped()
130 const T* fwHiddenState = context->getInputBuffer<T>(kFwHiddenStateTensor); in executeTyped()
132 const T* bwWeights = context->getInputBuffer<T>(kBwWeightsTensor); in executeTyped()
136 const T* bwBias = context->getInputBuffer<T>(kBwBiasTensor); in executeTyped()
137 const T* bwHiddenState = context->getInputBuffer<T>(kBwHiddenStateTensor); in executeTyped()
145 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
146 fwAuxWeights = context->getInputBuffer<T>(kFwAuxWeightsTensor); in executeTyped()
147 bwAuxWeights = context->getInputBuffer<T>(kBwAuxWeightsTensor); in executeTyped()
[all …]
H A DGenerateProposals.cpp291 return bboxTransformFloat32(context->getInputBuffer<float>(kRoiTensor), in execute()
597 const T_Roi* roiBase = context->getInputBuffer<T_Roi>(kRoiTensor); in boxWithNmsLimitWriteOutput()
820 context->getInputBuffer<_Float16>(kScoreTensor), in execute()
822 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
824 context->getInputBuffer<int32_t>(kBatchesTensor), in execute()
1697 context->getInputBuffer<_Float16>(kScoreTensor), in execute()
1699 context->getInputBuffer<_Float16>(kDeltaTensor), in execute()
1701 context->getInputBuffer<_Float16>(kAnchorTensor), in execute()
1725 context->getInputBuffer<float>(kScoreTensor), in execute()
1727 context->getInputBuffer<float>(kDeltaTensor), in execute()
[all …]
H A DStridedSlice.cpp92 context->getInputBuffer<T>(kInputTensor), context->getInputShape(kInputTensor), in executeTyped()
93 context->getInputBuffer<int32_t>(kBeginTensor), in executeTyped()
94 context->getInputBuffer<int32_t>(kEndTensor), in executeTyped()
95 context->getInputBuffer<int32_t>(kStridesTensor), in executeTyped()
163 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()
164 const int32_t* endData = context->getInputBuffer<int32_t>(kEndTensor); in prepare()
165 const int32_t* stridesData = context->getInputBuffer<int32_t>(kStridesTensor); in prepare()
H A DDepthwiseConv2D.cpp585 return depthwiseConv(context->getInputBuffer<float>(kInputTensor), in execute()
587 context->getInputBuffer<float>(kFilterTensor), in execute()
589 context->getInputBuffer<float>(kBiasTensor), in execute()
598 return depthwiseConv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
602 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
614 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
616 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
621 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
648 context->getInputBuffer<int8_t>(kInputTensor), in execute()
650 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
[all …]
H A DLogicalAndOr.cpp84 std::logical_and<bool>(), context->getInputBuffer<bool8>(kInputTensor1), in executeAnd()
85 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeAnd()
92 std::logical_or<bool>(), context->getInputBuffer<bool8>(kInputTensor1), in executeOr()
93 context->getInputShape(kInputTensor1), context->getInputBuffer<bool8>(kInputTensor2), in executeOr()
H A DQuantize.cpp101 return quantizeToQuant8<float>(context->getInputBuffer<float>(kInputTensor), in execute()
105 return quantizeToQuant8Signed<float>(context->getInputBuffer<float>(kInputTensor), in execute()
111 return quantizeToQuant8<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
115 return quantizeToQuant8Signed<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
H A DConv2D.cpp706 return conv(context->getInputBuffer<float>(kInputTensor), in execute()
708 context->getInputBuffer<float>(kFilterTensor), in execute()
710 context->getInputBuffer<float>(kBiasTensor), in execute()
718 return conv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
722 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
733 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
735 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
740 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
766 context->getInputBuffer<int8_t>(kInputTensor), in execute()
768 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
[all …]
H A DRoiAlign.cpp450 return roiAlign(context->getInputBuffer<_Float16>(kInputTensor), in execute()
452 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
454 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
464 return roiAlign(context->getInputBuffer<float>(kInputTensor), in execute()
466 context->getInputBuffer<float>(kRoiTensor), in execute()
468 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
478 return roiAlign(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
480 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
482 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
492 return roiAlign(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
H A DElementwise.cpp50 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
54 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
68 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
72 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
77 context->getInputBuffer<int32_t>(kInputTensor), in executeAbs()
H A DChannelShuffle.cpp98 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
102 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
106 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
110 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
H A DNeg.cpp72 return compute(context->getInputBuffer<_Float16>(kInputTensor), in execute()
76 return compute(context->getInputBuffer<float>(kInputTensor), in execute()
80 return compute(context->getInputBuffer<int32_t>(kInputTensor), in execute()
H A DHeatmapMaxKeypoint.cpp305 const auto heatmap = context->getInputBuffer<_Float16>(kHeatmapTensor); in execute()
307 const auto boxes = context->getInputBuffer<_Float16>(kBoxesTensor); in execute()
328 return heatmapMaxKeypointFloat32(context->getInputBuffer<float>(kHeatmapTensor), in execute()
330 context->getInputBuffer<float>(kBoxesTensor), in execute()
340 context->getInputBuffer<uint8_t>(kHeatmapTensor), in execute()
342 context->getInputBuffer<uint16_t>(kBoxesTensor), in execute()
351 context->getInputBuffer<int8_t>(kHeatmapTensor), in execute()
353 context->getInputBuffer<uint16_t>(kBoxesTensor), in execute()
H A DDequantize.cpp123 const uint8_t* inputBuffer = context->getInputBuffer<uint8_t>(kInputTensor); in execute()
131 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
139 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
147 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()

123