Home
last modified time | relevance | path

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

12

/packages/modules/NeuralNetworks/common/operations/
DGenerateProposals.cpp826 context->getInputValue<_Float16>(kScoreThresholdScalar), in execute()
827 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
828 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
829 context->getInputValue<_Float16>(kIoUThresholdScalar), in execute()
830 context->getInputValue<_Float16>(kSigmaScalar), in execute()
831 context->getInputValue<_Float16>(kNmsScoreThresholdScalar), in execute()
848 context->getInputValue<float>(kScoreThresholdScalar), in execute()
849 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
850 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
851 context->getInputValue<float>(kIoUThresholdScalar), in execute()
[all …]
DRoiAlign.cpp388 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
408 int32_t outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
409 int32_t outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
410 int32_t heightSamplingRatio = context->getInputValue<int32_t>(kHeightSamplingRatioScalar); in prepare()
411 int32_t widthSamplingRatio = context->getInputValue<int32_t>(kWidthSamplingRatioScalar); 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()
456 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
[all …]
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()
286 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
287 context->getInputValue<_Float16>(kWidthStrideScalar), in execute()
288 context->getInputValue<bool>(kLayoutScalar), in execute()
[all …]
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()
60 padding_bottom = context->getInputValue<int32_t>(4); in initialize()
61 stride_width = context->getInputValue<int32_t>(5); in initialize()
62 stride_height = context->getInputValue<int32_t>(6); in initialize()
63 filter_width = context->getInputValue<int32_t>(7); in initialize()
64 filter_height = context->getInputValue<int32_t>(8); in initialize()
65 activation = context->getInputValue<int32_t>(9); in initialize()
67 useNchw = context->getInputValue<bool>(10); in initialize()
[all …]
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()
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()
536 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()
586 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
[all …]
DDepthwiseConv2D.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 depth_multiplier = context->getInputValue<int32_t>(6); in initialize()
67 activation = context->getInputValue<int32_t>(7); in initialize()
69 useNchw = context->getInputValue<bool>(8); in initialize()
72 dilation_width_factor = context->getInputValue<int32_t>(9); in initialize()
73 dilation_height_factor = context->getInputValue<int32_t>(10); in initialize()
77 padding_left = context->getInputValue<int32_t>(3); in initialize()
78 padding_right = context->getInputValue<int32_t>(4); in initialize()
[all …]
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()
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 …]
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()
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()
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()
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()
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()
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()
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()
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()
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()
DElu.cpp84 context->getInputValue<_Float16>(kAlphaScalar), in execute()
90 context->getInputValue<float>(kAlphaScalar), in execute()
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), in concatenation()
187 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare()
DTopK_V2.cpp69 context->getInputValue<int32_t>(kTopKScalar), in executeTyped()
97 const int32_t k = context->getInputValue<int32_t>(kTopKScalar); in prepare()
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()
DGather.cpp84 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
102 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
/packages/modules/NeuralNetworks/runtime/test/fibonacci_extension/
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