Home
last modified time | relevance | path

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

12

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DRoiAlign.cpp388 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
409 int32_t outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
414 heightScale = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
415 widthScale = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
417 heightScale = context->getInputValue<float>(kHeightStrideSalar); in prepare()
418 widthScale = context->getInputValue<float>(kWidthStrideScalar); in prepare()
460 context->getInputValue<bool>(kLayoutScalar), in execute()
470 context->getInputValue<float>(kHeightStrideSalar), in execute()
474 context->getInputValue<bool>(kLayoutScalar), in execute()
488 context->getInputValue<bool>(kLayoutScalar), in execute()
[all …]
H A DRoiPooling.cpp231 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
246 auto outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
247 auto outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
250 heightStride = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
251 widthStride = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
253 heightStride = context->getInputValue<float>(kHeightStrideSalar); in prepare()
254 widthStride = context->getInputValue<float>(kWidthStrideScalar); in prepare()
288 context->getInputValue<bool>(kLayoutScalar), in execute()
300 context->getInputValue<bool>(kLayoutScalar), in execute()
312 context->getInputValue<bool>(kLayoutScalar), in execute()
[all …]
H A DGenerateProposals.cpp830 context->getInputValue<_Float16>(kSigmaScalar), in execute()
1290 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
1667 if (context->getInputValue<bool>(kUseRegularNmsScalar)) { in prepare()
1703 context->getInputValue<_Float16>(kScaleYScalar), in execute()
1704 context->getInputValue<_Float16>(kScaleXScalar), in execute()
1705 context->getInputValue<_Float16>(kScaleHScalar), in execute()
1706 context->getInputValue<_Float16>(kScaleWScalar), in execute()
1731 context->getInputValue<float>(kScaleYScalar), in execute()
1732 context->getInputValue<float>(kScaleXScalar), in execute()
1733 context->getInputValue<float>(kScaleHScalar), in execute()
[all …]
H A DInstanceNormalization.cpp138 context->getInputValue<_Float16>(kGammaScalar), in execute()
139 context->getInputValue<_Float16>(kBetaScalar), in execute()
140 context->getInputValue<_Float16>(kEpsilonScalar), in execute()
141 context->getInputValue<bool>(kLayoutScalar), in execute()
147 context->getInputValue<float>(kGammaScalar), in execute()
148 context->getInputValue<float>(kBetaScalar), in execute()
149 context->getInputValue<float>(kEpsilonScalar), in execute()
150 context->getInputValue<bool>(kLayoutScalar), in execute()
H A DPooling.cpp57 padding_left = context->getInputValue<int32_t>(1); in initialize()
58 padding_right = context->getInputValue<int32_t>(2); in initialize()
59 padding_top = context->getInputValue<int32_t>(3); in initialize()
61 stride_width = context->getInputValue<int32_t>(5); in initialize()
63 filter_width = context->getInputValue<int32_t>(7); in initialize()
65 activation = context->getInputValue<int32_t>(9); in initialize()
67 useNchw = context->getInputValue<bool>(10); in initialize()
71 stride_width = context->getInputValue<int32_t>(2); in initialize()
73 filter_width = context->getInputValue<int32_t>(4); in initialize()
75 activation = context->getInputValue<int32_t>(6); in initialize()
[all …]
H A DLocalResponseNormalization.cpp126 ? context->getInputValue<int32_t>(kAxisScalar) in executeTyped()
131 context->getInputValue<int32_t>(kRadiusScalar), context->getInputValue<T>(kBiasScalar), in executeTyped()
132 context->getInputValue<T>(kAlphaScalar), context->getInputValue<T>(kBetaScalar), axis, in executeTyped()
187 ? context->getInputValue<int32_t>(kAxisScalar) in prepare()
192 const int32_t radius = context->getInputValue<int32_t>(kRadiusScalar); in prepare()
H A DBroadcast.cpp504 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
512 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
520 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
528 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
554 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
562 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
570 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
578 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
604 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
620 context->getInputValue<int32_t>(kActivationScalar), in executeSub()
[all …]
H A DDepthwiseConv2D.cpp64 stride_width = context->getInputValue<int32_t>(4); in initialize()
65 stride_height = context->getInputValue<int32_t>(5); in initialize()
67 activation = context->getInputValue<int32_t>(7); in initialize()
69 useNchw = context->getInputValue<bool>(8); in initialize()
77 padding_left = context->getInputValue<int32_t>(3); in initialize()
78 padding_right = context->getInputValue<int32_t>(4); in initialize()
79 padding_top = context->getInputValue<int32_t>(5); in initialize()
81 stride_width = context->getInputValue<int32_t>(7); in initialize()
82 stride_height = context->getInputValue<int32_t>(8); in initialize()
84 activation = context->getInputValue<int32_t>(10); in initialize()
[all …]
H A DStridedSlice.cpp96 context->getInputValue<int32_t>(kBeginMask), context->getInputValue<int32_t>(kEndMask), in executeTyped()
97 context->getInputValue<int32_t>(kShrinkAxisMask), in executeTyped()
167 const int32_t beginMask = context->getInputValue<int32_t>(kBeginMask); in prepare()
168 const int32_t endMask = context->getInputValue<int32_t>(kEndMask); in prepare()
169 const int32_t shrinkAxisMask = context->getInputValue<int32_t>(kShrinkAxisMask); in prepare()
H A DChannelShuffle.cpp84 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in prepare()
85 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
93 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in execute()
94 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
H A DConv2D.cpp77 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()
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()
93 stride_width = context->getInputValue<int32_t>(7); in initialize()
94 stride_height = context->getInputValue<int32_t>(8); in initialize()
95 activation = context->getInputValue<int32_t>(9); in initialize()
[all …]
H A DTransposeConv2D.cpp71 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()
92 strideWidth = context->getInputValue<int32_t>(7); in initialize()
93 strideHeight = context->getInputValue<int32_t>(8); in initialize()
94 activation = context->getInputValue<int32_t>(9); in initialize()
[all …]
H A DSoftmax.cpp275 ? context->getInputValue<_Float16>(kBetaScalar) in prepare()
276 : context->getInputValue<float>(kBetaScalar); in prepare()
288 ? context->getInputValue<int32_t>(kAxisScalar) in execute()
294 context->getInputValue<_Float16>(kBetaScalar), axis, in execute()
300 context->getInputValue<float>(kBetaScalar), axis, in execute()
306 context->getInputValue<float>(kBetaScalar), axis, in execute()
312 context->getInputValue<float>(kBetaScalar), axis, in execute()
H A DResizeImageOps.cpp169 scalarValue = context->getInputValue<bool>(scalarIndex); in getOptionalScalar()
252 height = context->getInputValue<int32_t>(kOutputHeightParamScalar); in prepare()
253 width = context->getInputValue<int32_t>(kOutputWidthParamScalar); in prepare()
256 context->getInputValue<float>(kOutputHeightParamScalar)); in prepare()
258 context->getInputValue<float>(kOutputWidthParamScalar)); in prepare()
262 static_cast<float>(context->getInputValue<_Float16>(kOutputHeightParamScalar))); in prepare()
265 static_cast<float>(context->getInputValue<_Float16>(kOutputWidthParamScalar))); in prepare()
H A 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()
H A DElu.cpp84 context->getInputValue<_Float16>(kAlphaScalar), in execute()
90 context->getInputValue<float>(kAlphaScalar), in execute()
H A DFullyConnected.cpp315 context->getInputValue<int32_t>(kActivationScalar), in execute()
325 context->getInputValue<int32_t>(kActivationScalar), in execute()
335 context->getInputValue<int32_t>(kActivationScalar), in execute()
345 context->getInputValue<int32_t>(kActivationScalar), in execute()
H A DUnidirectionalSequenceRNN.cpp74 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
79 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped()
160 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare()
H A DQLSTM.cpp397 const float cellClip = context->getInputValue<float>(kCellClip); in execute()
398 const float projectionClip = context->getInputValue<float>(kProjectionClip); in execute()
399 const float inputIntermediateScale = context->getInputValue<float>(kInputIntermediateScale); in execute()
400 const float forgetIntermediateScale = context->getInputValue<float>(kForgetIntermediateScale); in execute()
401 const float cellIntermediateScale = context->getInputValue<float>(kCellIntermediateScale); in execute()
402 const float outputIntermediateScale = context->getInputValue<float>(kOutputIntermediateScale); in execute()
403 const int8_t hiddenStateZeroPoint = context->getInputValue<int8_t>(kHiddenStateZeroPoint); in execute()
404 const float hiddenStateScale = context->getInputValue<float>(kHiddenStateScale); in execute()
H A DBidirectionalSequenceRNN.cpp158 const int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
159 const bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped()
160 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
348 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
384 bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare()
H A DConcatenation.cpp106 return concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
133 NN_RET_CHECK(concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount),
187 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare()
H A DTopK_V2.cpp69 context->getInputValue<int32_t>(kTopKScalar), in executeTyped()
97 const int32_t k = context->getInputValue<int32_t>(kTopKScalar); in prepare()
H A DGather.cpp84 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
102 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
H A DUnidirectionalSequenceLSTM.cpp99 return context->getInputValue<bool>(kTimeMajorParam); in isTimeMajor()
106 static_cast<TfLiteFusedActivation>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
107 params.cell_clip = static_cast<float>(context->getInputValue<T>(kCellClipParam)); in getLSTMParams()
108 params.proj_clip = static_cast<float>(context->getInputValue<T>(kProjClipParam)); in getLSTMParams()
/aosp12/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
H A DFibonacciDriver.cpp83 n = static_cast<int64_t>(context->getInputValue<float>(kInputN)); in prepare()
85 n = context->getInputValue<int64_t>(kInputN); in prepare()
119 n = static_cast<int64_t>(context->getInputValue<float>(kInputN)); in execute()
121 n = context->getInputValue<int64_t>(kInputN); in execute()

12