Home
last modified time | relevance | path

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

123

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp94 return context->getInputBuffer(tensor) != nullptr; in hasTensor()
423 getLSTMParams<float>(context), context->getInputBuffer<float>(kInputTensor), in execute()
425 context->getInputBuffer<float>(kInputToInputWeightsTensor), in execute()
426 context->getInputBuffer<float>(kInputToForgetWeightsTensor), in execute()
427 context->getInputBuffer<float>(kInputToCellWeightsTensor), in execute()
428 context->getInputBuffer<float>(kInputToOutputWeightsTensor), in execute()
430 context->getInputBuffer<float>(kRecurrentToInputWeightsTensor), in execute()
431 context->getInputBuffer<float>(kRecurrentToForgetWeightsTensor), in execute()
432 context->getInputBuffer<float>(kRecurrentToCellWeightsTensor), in execute()
433 context->getInputBuffer<float>(kRecurrentToOutputWeightsTensor), in execute()
[all …]
DSlice.cpp118 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()
119 const int32_t* sizeData = context->getInputBuffer<int32_t>(kSizeTensor); in prepare()
142 return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
144 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
146 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
151 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
153 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
155 context->getInputBuffer<int32_t>(kSizeTensor), in execute()
160 return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor), in execute()
162 context->getInputBuffer<int32_t>(kBeginTensor), in execute()
[all …]
DGather.cpp109 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
111 context->getInputBuffer<int32_t>(kInputIndices), in execute()
115 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
117 context->getInputBuffer<int32_t>(kInputIndices), in execute()
121 return eval(context->getInputBuffer<int32_t>(kInputTensor), in execute()
123 context->getInputBuffer<int32_t>(kInputIndices), in execute()
127 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
129 context->getInputBuffer<int32_t>(kInputIndices), in execute()
133 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
135 context->getInputBuffer<int32_t>(kInputIndices), in execute()
DBroadcast.cpp494 return addFloat16(context->getInputBuffer<_Float16>(kInputTensor1), in executeAdd()
496 context->getInputBuffer<_Float16>(kInputTensor2), in executeAdd()
502 return addFloat32(context->getInputBuffer<float>(kInputTensor1), in executeAdd()
504 context->getInputBuffer<float>(kInputTensor2), in executeAdd()
510 return addQuant8(context->getInputBuffer<uint8_t>(kInputTensor1), in executeAdd()
512 context->getInputBuffer<uint8_t>(kInputTensor2), in executeAdd()
518 return addQuant8(context->getInputBuffer<int8_t>(kInputTensor1), in executeAdd()
520 context->getInputBuffer<int8_t>(kInputTensor2), in executeAdd()
526 return executeInt32(context->getInputBuffer<int32_t>(kInputTensor1), in executeAdd()
528 context->getInputBuffer<int32_t>(kInputTensor2), in executeAdd()
[all …]
DTranspose.cpp116 const int32_t* permData = context->getInputBuffer<int32_t>(kPermTensor); in prepare()
142 return transposeGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
144 context->getInputBuffer<int32_t>(kPermTensor), in execute()
149 return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
151 context->getInputBuffer<int32_t>(kPermTensor), in execute()
156 return transposeGeneric(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
158 context->getInputBuffer<int32_t>(kPermTensor), in execute()
163 return transposeGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
165 context->getInputBuffer<int32_t>(kPermTensor), in execute()
DActivation.cpp438 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
443 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
448 return reluQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu()
453 return reluQuant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu()
467 return relu1Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu1()
472 return relu1Float(context->getInputBuffer<float>(kInputTensor), in executeRelu1()
477 return relu1Quant8(context->getInputBuffer<uint8_t>(kInputTensor), in executeRelu1()
482 return relu1Quant8Signed(context->getInputBuffer<int8_t>(kInputTensor), in executeRelu1()
496 return relu6Float(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu6()
501 return relu6Float(context->getInputBuffer<float>(kInputTensor), in executeRelu6()
[all …]
DComparisons.cpp76 std::less<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessTyped()
77 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessTyped()
85 std::less_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeLessEqualTyped()
86 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeLessEqualTyped()
94 std::equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeEqualTyped()
95 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeEqualTyped()
103 std::not_equal_to<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeNotEqualTyped()
104 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeNotEqualTyped()
112 std::greater_equal<ComparisonType>(), context->getInputBuffer<DataType>(kInputTensor1), in executeGreaterEqualTyped()
113 context->getInputShape(kInputTensor1), context->getInputBuffer<DataType>(kInputTensor2), in executeGreaterEqualTyped()
[all …]
DPRelu.cpp135 context->getInputBuffer<_Float16>(kInputTensor), in execute()
137 context->getInputBuffer<_Float16>(kAlphaTensor), in execute()
146 context->getInputBuffer<float>(kInputTensor), in execute()
148 context->getInputBuffer<float>(kAlphaTensor), in execute()
153 return evalQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
155 context->getInputBuffer<uint8_t>(kAlphaTensor), in execute()
161 return evalQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
163 context->getInputBuffer<int8_t>(kAlphaTensor), in execute()
DFullyConnected.cpp306 return fullyConnectedFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
308 context->getInputBuffer<float>(kWeightsTensor), in execute()
310 context->getInputBuffer<float>(kBiasTensor), in execute()
316 return fullyConnectedFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
318 context->getInputBuffer<_Float16>(kWeightsTensor), in execute()
320 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
326 return fullyConnectedQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
328 context->getInputBuffer<uint8_t>(kWeightsTensor), in execute()
330 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
336 return fullyConnectedQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
DQLSTM.cpp99 return context->getInputBuffer(tensor) != nullptr; in hasTensor()
405 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputTensor)); in execute()
408 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToInputWeightsTensor)); in execute()
411 context->getInputBuffer(kRecurrentToInputWeightsTensor)); in execute()
413 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToInputWeightsTensor)); in execute()
415 reinterpret_cast<const int16_t*>(context->getInputBuffer(kInputLayerNormTensor)); in execute()
417 reinterpret_cast<const int32_t*>(context->getInputBuffer(kInputGateBiasTensor)); in execute()
420 reinterpret_cast<const int8_t*>(context->getInputBuffer(kInputToForgetWeightsTensor)); in execute()
422 context->getInputBuffer(kRecurrentToForgetWeightsTensor)); in execute()
424 reinterpret_cast<const int16_t*>(context->getInputBuffer(kCellToForgetWeightsTensor)); in execute()
[all …]
DRoiPooling.cpp279 return roiPooling(context->getInputBuffer<_Float16>(kInputTensor), in execute()
281 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
283 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
291 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
293 context->getInputBuffer<float>(kRoiTensor), in execute()
295 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
303 return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
305 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
307 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
315 return roiPooling(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
DTransposeConv2D.cpp80 const int32_t* outputShapeData = context->getInputBuffer<int32_t>(3); in initialize()
548 return transposeConv(context->getInputBuffer<float>(kInputTensor), in execute()
550 context->getInputBuffer<float>(kFilterTensor), in execute()
552 context->getInputBuffer<float>(kBiasTensor), in execute()
557 return transposeConv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
559 context->getInputBuffer<_Float16>(kFilterTensor), in execute()
561 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
569 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
571 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
574 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
[all …]
DBidirectionalSequenceRNN.cpp124 const T* input = context->getInputBuffer<T>(kInputTensor); in executeTyped()
127 const T* fwWeights = context->getInputBuffer<T>(kFwWeightsTensor); in executeTyped()
129 const T* fwRecurrentWeights = context->getInputBuffer<T>(kFwRecurrentWeightsTensor); in executeTyped()
131 const T* fwBias = context->getInputBuffer<T>(kFwBiasTensor); in executeTyped()
132 const T* fwHiddenState = context->getInputBuffer<T>(kFwHiddenStateTensor); in executeTyped()
134 const T* bwWeights = context->getInputBuffer<T>(kBwWeightsTensor); in executeTyped()
136 const T* bwRecurrentWeights = context->getInputBuffer<T>(kBwRecurrentWeightsTensor); in executeTyped()
138 const T* bwBias = context->getInputBuffer<T>(kBwBiasTensor); in executeTyped()
139 const T* bwHiddenState = context->getInputBuffer<T>(kBwHiddenStateTensor); in executeTyped()
147 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
[all …]
DStridedSlice.cpp91 context->getInputBuffer<T>(kInputTensor), context->getInputShape(kInputTensor), in executeTyped()
92 context->getInputBuffer<int32_t>(kBeginTensor), in executeTyped()
93 context->getInputBuffer<int32_t>(kEndTensor), in executeTyped()
94 context->getInputBuffer<int32_t>(kStridesTensor), in executeTyped()
160 const int32_t* beginData = context->getInputBuffer<int32_t>(kBeginTensor); in prepare()
161 const int32_t* endData = context->getInputBuffer<int32_t>(kEndTensor); in prepare()
162 const int32_t* stridesData = context->getInputBuffer<int32_t>(kStridesTensor); in prepare()
DGenerateProposals.cpp278 return bboxTransformFloat16(context->getInputBuffer<_Float16>(kRoiTensor), in execute()
280 context->getInputBuffer<_Float16>(kDeltaTensor), in execute()
282 context->getInputBuffer<int32_t>(kBatchesTensor), in execute()
284 context->getInputBuffer<_Float16>(kImageInfoTensor), in execute()
290 return bboxTransformFloat32(context->getInputBuffer<float>(kRoiTensor), in execute()
292 context->getInputBuffer<float>(kDeltaTensor), in execute()
294 context->getInputBuffer<int32_t>(kBatchesTensor), in execute()
296 context->getInputBuffer<float>(kImageInfoTensor), in execute()
303 return bboxTransformQuant(context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
305 context->getInputBuffer<uint8_t>(kDeltaTensor), in execute()
[all …]
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()
DDepthwiseConv2D.cpp577 return depthwiseConv(context->getInputBuffer<float>(kInputTensor), in execute()
579 context->getInputBuffer<float>(kFilterTensor), in execute()
581 context->getInputBuffer<float>(kBiasTensor), in execute()
590 return depthwiseConv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
592 context->getInputBuffer<_Float16>(kFilterTensor), in execute()
594 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
606 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
608 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
611 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
619 return depthwiseConv(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
DQuantize.cpp104 return quantizeToQuant8<float>(context->getInputBuffer<float>(kInputTensor), in execute()
108 return quantizeToQuant8Signed<float>(context->getInputBuffer<float>(kInputTensor), in execute()
114 return quantizeToQuant8<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
118 return quantizeToQuant8Signed<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
DNeg.cpp75 return compute(context->getInputBuffer<_Float16>(kInputTensor), in execute()
79 return compute(context->getInputBuffer<float>(kInputTensor), in execute()
83 return compute(context->getInputBuffer<int32_t>(kInputTensor), in execute()
DChannelShuffle.cpp101 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
105 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
109 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
113 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
DElementwise.cpp53 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
57 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
71 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
75 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
80 context->getInputBuffer<int32_t>(kInputTensor), in executeAbs()
DRoiAlign.cpp447 return roiAlign(context->getInputBuffer<_Float16>(kInputTensor), in execute()
449 context->getInputBuffer<_Float16>(kRoiTensor), in execute()
451 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
461 return roiAlign(context->getInputBuffer<float>(kInputTensor), in execute()
463 context->getInputBuffer<float>(kRoiTensor), in execute()
465 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
475 return roiAlign(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
477 context->getInputBuffer<uint16_t>(kRoiTensor), in execute()
479 context->getInputBuffer<int32_t>(kBatchSplitTensor), in execute()
489 return roiAlign(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
DConv2D.cpp699 return conv(context->getInputBuffer<float>(kInputTensor), in execute()
701 context->getInputBuffer<float>(kFilterTensor), in execute()
703 context->getInputBuffer<float>(kBiasTensor), in execute()
711 return conv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
713 context->getInputBuffer<_Float16>(kFilterTensor), in execute()
715 context->getInputBuffer<_Float16>(kBiasTensor), in execute()
726 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
728 context->getInputBuffer<int8_t>(kFilterTensor), in execute()
731 context->getInputBuffer<int32_t>(kBiasTensor), in execute()
739 return conv(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
DDequantize.cpp124 const uint8_t* inputBuffer = context->getInputBuffer<uint8_t>(kInputTensor); in execute()
132 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
140 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
148 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
DHeatmapMaxKeypoint.cpp303 const auto heatmap = context->getInputBuffer<_Float16>(kHeatmapTensor); in execute()
305 const auto boxes = context->getInputBuffer<_Float16>(kBoxesTensor); in execute()
326 return heatmapMaxKeypointFloat32(context->getInputBuffer<float>(kHeatmapTensor), in execute()
328 context->getInputBuffer<float>(kBoxesTensor), in execute()
338 context->getInputBuffer<uint8_t>(kHeatmapTensor), in execute()
340 context->getInputBuffer<uint16_t>(kBoxesTensor), in execute()
349 context->getInputBuffer<int8_t>(kHeatmapTensor), in execute()
351 context->getInputBuffer<uint16_t>(kBoxesTensor), in execute()

123