Home
last modified time | relevance | path

Searched refs:getInputValue (Results 1 – 23 of 23) sorted by relevance

/frameworks/ml/nn/common/operations/
DRoiPooling.cpp203 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
218 auto outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
219 auto outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
222 heightStride = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
223 widthStride = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
225 heightStride = context->getInputValue<float>(kHeightStrideSalar); in prepare()
226 widthStride = context->getInputValue<float>(kWidthStrideScalar); in prepare()
258 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
259 context->getInputValue<_Float16>(kWidthStrideScalar), in execute()
260 context->getInputValue<bool>(kLayoutScalar), in execute()
[all …]
DRoiAlign.cpp368 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
388 int32_t outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
389 int32_t outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
390 int32_t heightSamplingRatio = context->getInputValue<int32_t>(kHeightSamplingRatioScalar); in prepare()
391 int32_t widthSamplingRatio = context->getInputValue<int32_t>(kWidthSamplingRatioScalar); in prepare()
394 heightScale = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
395 widthScale = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
397 heightScale = context->getInputValue<float>(kHeightStrideSalar); in prepare()
398 widthScale = context->getInputValue<float>(kWidthStrideScalar); in prepare()
436 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
[all …]
DGenerateProposals.cpp734 context->getInputValue<_Float16>(kScoreThresholdScalar), in execute()
735 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
736 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
737 context->getInputValue<_Float16>(kIoUThresholdScalar), in execute()
738 context->getInputValue<_Float16>(kSigmaScalar), in execute()
739 context->getInputValue<_Float16>(kNmsScoreThresholdScalar), in execute()
756 context->getInputValue<float>(kScoreThresholdScalar), in execute()
757 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
758 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
759 context->getInputValue<float>(kIoUThresholdScalar), in execute()
[all …]
DInstanceNormalization.cpp123 context->getInputValue<_Float16>(kGammaScalar), in execute()
124 context->getInputValue<_Float16>(kBetaScalar), in execute()
125 context->getInputValue<_Float16>(kEpsilonScalar), in execute()
126 context->getInputValue<bool>(kLayoutScalar), in execute()
132 context->getInputValue<float>(kGammaScalar), in execute()
133 context->getInputValue<float>(kBetaScalar), in execute()
134 context->getInputValue<float>(kEpsilonScalar), in execute()
135 context->getInputValue<bool>(kLayoutScalar), in execute()
DPooling.cpp47 padding_left = context->getInputValue<int32_t>(1); in initialize()
48 padding_right = context->getInputValue<int32_t>(2); in initialize()
49 padding_top = context->getInputValue<int32_t>(3); in initialize()
50 padding_bottom = context->getInputValue<int32_t>(4); in initialize()
51 stride_width = context->getInputValue<int32_t>(5); in initialize()
52 stride_height = context->getInputValue<int32_t>(6); in initialize()
53 filter_width = context->getInputValue<int32_t>(7); in initialize()
54 filter_height = context->getInputValue<int32_t>(8); in initialize()
55 activation = context->getInputValue<int32_t>(9); in initialize()
57 useNchw = context->getInputValue<bool>(10); in initialize()
[all …]
DResizeImageOps.cpp148 useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
163 height = context->getInputValue<int32_t>(kOutputHeightParamScalar); in prepare()
164 width = context->getInputValue<int32_t>(kOutputWidthParamScalar); in prepare()
167 context->getInputValue<float>(kOutputHeightParamScalar)); in prepare()
169 context->getInputValue<float>(kOutputWidthParamScalar)); in prepare()
173 static_cast<float>(context->getInputValue<_Float16>(kOutputHeightParamScalar))); in prepare()
176 static_cast<float>(context->getInputValue<_Float16>(kOutputWidthParamScalar))); in prepare()
197 useNchw = context->getInputValue<bool>(kLayoutScalar); in execute()
DConv2D.cpp63 padding_implicit = context->getInputValue<int32_t>(3); in initialize()
64 stride_width = context->getInputValue<int32_t>(4); in initialize()
65 stride_height = context->getInputValue<int32_t>(5); in initialize()
66 activation = context->getInputValue<int32_t>(6); in initialize()
68 useNchw = context->getInputValue<bool>(7); in initialize()
71 dilation_width_factor = context->getInputValue<int32_t>(8); in initialize()
72 dilation_height_factor = context->getInputValue<int32_t>(9); in initialize()
76 padding_left = context->getInputValue<int32_t>(3); in initialize()
77 padding_right = context->getInputValue<int32_t>(4); in initialize()
78 padding_top = context->getInputValue<int32_t>(5); in initialize()
[all …]
DChannelShuffle.cpp76 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in prepare()
77 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
85 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in execute()
86 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DTransposeConv2D.cpp60 paddingImplicit = context->getInputValue<int32_t>(4); in initialize()
61 strideWidth = context->getInputValue<int32_t>(5); in initialize()
62 strideHeight = context->getInputValue<int32_t>(6); in initialize()
63 activation = context->getInputValue<int32_t>(7); in initialize()
64 useNchw = context->getInputValue<bool>(8); in initialize()
78 paddingLeft = context->getInputValue<int32_t>(3); in initialize()
79 paddingRight = context->getInputValue<int32_t>(4); in initialize()
80 paddingTop = context->getInputValue<int32_t>(5); in initialize()
81 paddingBottom = context->getInputValue<int32_t>(6); in initialize()
82 strideWidth = context->getInputValue<int32_t>(7); in initialize()
[all …]
DBroadcast.cpp401 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
409 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
417 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
434 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
442 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
450 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
467 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
475 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
483 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
500 context->getInputValue<int32_t>(kActivationScalar), in executeDiv()
[all …]
DSoftmax.cpp257 ? context->getInputValue<_Float16>(kBetaScalar) in prepare()
258 : context->getInputValue<float>(kBetaScalar); in prepare()
270 ? context->getInputValue<int32_t>(kAxisScalar) in execute()
276 context->getInputValue<_Float16>(kBetaScalar), axis, in execute()
282 context->getInputValue<float>(kBetaScalar), axis, in execute()
288 context->getInputValue<float>(kBetaScalar), axis, in execute()
DLogSoftmax.cpp98 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
104 context->getInputValue<_Float16>(kInputBeta), axis, in execute()
109 context->getInputValue<float>(kInputBeta), axis, in execute()
DUnidirectionalSequenceRNN.cpp66 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
71 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped()
138 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare()
DBidirectionalSequenceRNN.cpp109 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
110 int32_t timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped()
111 int32_t mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
273 int32_t mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
314 int32_t timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare()
DFullyConnected.cpp237 context->getInputValue<int32_t>(kActivationScalar), in execute()
247 context->getInputValue<int32_t>(kActivationScalar), in execute()
257 context->getInputValue<int32_t>(kActivationScalar), in execute()
DGather.cpp79 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
97 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DConcatenation.cpp93 return concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
132 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare()
DUnidirectionalSequenceLSTM.cpp91 return context->getInputValue<bool>(kTimeMajorParam); in isTimeMajor()
98 static_cast<TfLiteFusedActivation>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
99 params.cell_clip = static_cast<float>(context->getInputValue<T>(kCellClipParam)); in getLSTMParams()
100 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
DL2Normalization.cpp173 ? context->getInputValue<int32_t>(kAxisScalar) in prepare()
192 ? context->getInputValue<int32_t>(kAxisScalar) in execute()
DReduce.cpp60 context->getInputValue<bool8>(kInputKeepDims), tempIndex.data(), tempAxes.data(), init, in compute()
123 bool keepDims = context->getInputValue<bool8>(kInputKeepDims); in prepare()
DHeatmapMaxKeypoint.cpp225 bool layout = context->getInputValue<bool>(kLayoutScalar); in prepare()
260 bool layout = context->getInputValue<bool>(kLayoutScalar); in execute()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp81 n = static_cast<int64_t>(context->getInputValue<float>(kInputN)); in prepare()
83 n = context->getInputValue<int64_t>(kInputN); in prepare()
117 n = static_cast<int64_t>(context->getInputValue<float>(kInputN)); in execute()
119 n = context->getInputValue<int64_t>(kInputN); in execute()
/frameworks/ml/nn/common/include/
DOperationsUtils.h117 T getInputValue(uint32_t index) const { in getInputValue() function