Home
last modified time | relevance | path

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

12

/frameworks/ml/nn/common/operations/
DRoiAlign.cpp385 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
405 int32_t outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
406 int32_t outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
407 int32_t heightSamplingRatio = context->getInputValue<int32_t>(kHeightSamplingRatioScalar); in prepare()
408 int32_t widthSamplingRatio = context->getInputValue<int32_t>(kWidthSamplingRatioScalar); in prepare()
411 heightScale = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
412 widthScale = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
414 heightScale = context->getInputValue<float>(kHeightStrideSalar); in prepare()
415 widthScale = context->getInputValue<float>(kWidthStrideScalar); in prepare()
453 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
[all …]
DRoiPooling.cpp230 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
245 auto outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
246 auto outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
249 heightStride = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
250 widthStride = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
252 heightStride = context->getInputValue<float>(kHeightStrideSalar); in prepare()
253 widthStride = context->getInputValue<float>(kWidthStrideScalar); in prepare()
285 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
286 context->getInputValue<_Float16>(kWidthStrideScalar), in execute()
287 context->getInputValue<bool>(kLayoutScalar), in execute()
[all …]
DGenerateProposals.cpp821 context->getInputValue<_Float16>(kScoreThresholdScalar), in execute()
822 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
823 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
824 context->getInputValue<_Float16>(kIoUThresholdScalar), in execute()
825 context->getInputValue<_Float16>(kSigmaScalar), in execute()
826 context->getInputValue<_Float16>(kNmsScoreThresholdScalar), in execute()
843 context->getInputValue<float>(kScoreThresholdScalar), in execute()
844 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
845 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
846 context->getInputValue<float>(kIoUThresholdScalar), in execute()
[all …]
DInstanceNormalization.cpp136 context->getInputValue<_Float16>(kGammaScalar), in execute()
137 context->getInputValue<_Float16>(kBetaScalar), in execute()
138 context->getInputValue<_Float16>(kEpsilonScalar), in execute()
139 context->getInputValue<bool>(kLayoutScalar), in execute()
145 context->getInputValue<float>(kGammaScalar), in execute()
146 context->getInputValue<float>(kBetaScalar), in execute()
147 context->getInputValue<float>(kEpsilonScalar), in execute()
148 context->getInputValue<bool>(kLayoutScalar), in execute()
DPooling.cpp55 padding_left = context->getInputValue<int32_t>(1); in initialize()
56 padding_right = context->getInputValue<int32_t>(2); in initialize()
57 padding_top = context->getInputValue<int32_t>(3); in initialize()
58 padding_bottom = context->getInputValue<int32_t>(4); in initialize()
59 stride_width = context->getInputValue<int32_t>(5); in initialize()
60 stride_height = context->getInputValue<int32_t>(6); in initialize()
61 filter_width = context->getInputValue<int32_t>(7); in initialize()
62 filter_height = context->getInputValue<int32_t>(8); in initialize()
63 activation = context->getInputValue<int32_t>(9); in initialize()
65 useNchw = context->getInputValue<bool>(10); in initialize()
[all …]
DLocalResponseNormalization.cpp124 ? context->getInputValue<int32_t>(kAxisScalar) in executeTyped()
129 context->getInputValue<int32_t>(kRadiusScalar), context->getInputValue<T>(kBiasScalar), in executeTyped()
130 context->getInputValue<T>(kAlphaScalar), context->getInputValue<T>(kBetaScalar), axis, in executeTyped()
181 ? context->getInputValue<int32_t>(kAxisScalar) in prepare()
DStridedSlice.cpp95 context->getInputValue<int32_t>(kBeginMask), context->getInputValue<int32_t>(kEndMask), in executeTyped()
96 context->getInputValue<int32_t>(kShrinkAxisMask), in executeTyped()
164 const int32_t beginMask = context->getInputValue<int32_t>(kBeginMask); in prepare()
165 const int32_t endMask = context->getInputValue<int32_t>(kEndMask); in prepare()
166 const int32_t shrinkAxisMask = context->getInputValue<int32_t>(kShrinkAxisMask); in prepare()
DBroadcast.cpp498 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
506 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
514 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
522 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
530 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
548 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
556 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
564 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
572 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
580 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
[all …]
DChannelShuffle.cpp87 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in prepare()
88 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
96 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in execute()
97 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DSoftmax.cpp268 ? context->getInputValue<_Float16>(kBetaScalar) in prepare()
269 : context->getInputValue<float>(kBetaScalar); in prepare()
281 ? context->getInputValue<int32_t>(kAxisScalar) in execute()
287 context->getInputValue<_Float16>(kBetaScalar), axis, in execute()
293 context->getInputValue<float>(kBetaScalar), axis, in execute()
299 context->getInputValue<float>(kBetaScalar), axis, in execute()
305 context->getInputValue<float>(kBetaScalar), axis, in execute()
DConv2D.cpp76 padding_implicit = context->getInputValue<int32_t>(3); in initialize()
77 stride_width = context->getInputValue<int32_t>(4); in initialize()
78 stride_height = context->getInputValue<int32_t>(5); in initialize()
79 activation = context->getInputValue<int32_t>(6); in initialize()
81 useNchw = context->getInputValue<bool>(7); in initialize()
84 dilation_width_factor = context->getInputValue<int32_t>(8); in initialize()
85 dilation_height_factor = context->getInputValue<int32_t>(9); in initialize()
89 padding_left = context->getInputValue<int32_t>(3); in initialize()
90 padding_right = context->getInputValue<int32_t>(4); in initialize()
91 padding_top = context->getInputValue<int32_t>(5); in initialize()
[all …]
DTransposeConv2D.cpp70 paddingImplicit = context->getInputValue<int32_t>(4); in initialize()
71 strideWidth = context->getInputValue<int32_t>(5); in initialize()
72 strideHeight = context->getInputValue<int32_t>(6); in initialize()
73 activation = context->getInputValue<int32_t>(7); in initialize()
74 useNchw = context->getInputValue<bool>(8); in initialize()
88 paddingLeft = context->getInputValue<int32_t>(3); in initialize()
89 paddingRight = context->getInputValue<int32_t>(4); in initialize()
90 paddingTop = context->getInputValue<int32_t>(5); in initialize()
91 paddingBottom = context->getInputValue<int32_t>(6); in initialize()
92 strideWidth = context->getInputValue<int32_t>(7); in initialize()
[all …]
DResizeImageOps.cpp167 scalarValue = context->getInputValue<bool>(scalarIndex); in getOptionalScalar()
246 height = context->getInputValue<int32_t>(kOutputHeightParamScalar); in prepare()
247 width = context->getInputValue<int32_t>(kOutputWidthParamScalar); in prepare()
250 context->getInputValue<float>(kOutputHeightParamScalar)); in prepare()
252 context->getInputValue<float>(kOutputWidthParamScalar)); in prepare()
256 static_cast<float>(context->getInputValue<_Float16>(kOutputHeightParamScalar))); in prepare()
259 static_cast<float>(context->getInputValue<_Float16>(kOutputWidthParamScalar))); in prepare()
DLogSoftmax.cpp100 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
106 context->getInputValue<_Float16>(kInputBeta), axis, in execute()
111 context->getInputValue<float>(kInputBeta), axis, in execute()
DElu.cpp85 context->getInputValue<_Float16>(kAlphaScalar), in execute()
91 context->getInputValue<float>(kAlphaScalar), in execute()
DFullyConnected.cpp312 context->getInputValue<int32_t>(kActivationScalar), in execute()
322 context->getInputValue<int32_t>(kActivationScalar), in execute()
332 context->getInputValue<int32_t>(kActivationScalar), in execute()
342 context->getInputValue<int32_t>(kActivationScalar), in execute()
DUnidirectionalSequenceRNN.cpp75 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
80 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped()
160 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare()
DTopK_V2.cpp72 context->getInputValue<int32_t>(kTopKScalar), in executeTyped()
100 const int32_t k = context->getInputValue<int32_t>(kTopKScalar); in prepare()
DConcatenation.cpp104 return concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
131 NN_RET_CHECK(concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
181 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare()
DBidirectionalSequenceRNN.cpp160 const int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
161 const bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped()
162 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
349 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
385 bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare()
DQLSTM.cpp395 const float cellClip = context->getInputValue<float>(kCellClip); in execute()
396 const float projectionClip = context->getInputValue<float>(kProjectionClip); in execute()
397 const float inputIntermediateScale = context->getInputValue<float>(kInputIntermediateScale); in execute()
398 const float forgetIntermediateScale = context->getInputValue<float>(kForgetIntermediateScale); in execute()
399 const float cellIntermediateScale = context->getInputValue<float>(kCellIntermediateScale); in execute()
400 const float outputIntermediateScale = context->getInputValue<float>(kOutputIntermediateScale); in execute()
401 const int8_t hiddenStateZeroPoint = context->getInputValue<int8_t>(kHiddenStateZeroPoint); in execute()
402 const float hiddenStateScale = context->getInputValue<float>(kHiddenStateScale); in execute()
DGather.cpp87 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
105 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DUnidirectionalSequenceLSTM.cpp98 return context->getInputValue<bool>(kTimeMajorParam); in isTimeMajor()
105 static_cast<TfLiteFusedActivation>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
106 params.cell_clip = static_cast<float>(context->getInputValue<T>(kCellClipParam)); in getLSTMParams()
107 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
DFill.cpp42 const T value = context->getInputValue<T>(kValueScalar); in executeTyped()
/frameworks/ml/nn/runtime/test/fibonacci_extension/
DFibonacciDriver.cpp84 n = static_cast<int64_t>(context->getInputValue<float>(kInputN)); in prepare()
86 n = context->getInputValue<int64_t>(kInputN); in prepare()
120 n = static_cast<int64_t>(context->getInputValue<float>(kInputN)); in execute()
122 n = context->getInputValue<int64_t>(kInputN); in execute()

12