Home
last modified time | relevance | path

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

12

/packages/modules/NeuralNetworks/common/cpu_operations/
DGenerateProposals.cpp733 context->getInputValue<_Float16>(kScoreThresholdScalar), in execute()
734 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
735 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
736 context->getInputValue<_Float16>(kIoUThresholdScalar), in execute()
737 context->getInputValue<_Float16>(kSigmaScalar), in execute()
738 context->getInputValue<_Float16>(kNmsScoreThresholdScalar), in execute()
755 context->getInputValue<float>(kScoreThresholdScalar), in execute()
756 context->getInputValue<int32_t>(kMaxNumDetectionScalar), in execute()
757 context->getInputValue<int32_t>(kNmsKernelScalar), in execute()
758 context->getInputValue<float>(kIoUThresholdScalar), in execute()
[all …]
DRoiAlign.cpp333 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
353 int32_t outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
354 int32_t outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
355 int32_t heightSamplingRatio = context->getInputValue<int32_t>(kHeightSamplingRatioScalar); in prepare()
356 int32_t widthSamplingRatio = context->getInputValue<int32_t>(kWidthSamplingRatioScalar); in prepare()
359 heightScale = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
360 widthScale = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
362 heightScale = context->getInputValue<float>(kHeightStrideSalar); in prepare()
363 widthScale = context->getInputValue<float>(kWidthStrideScalar); in prepare()
401 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
[all …]
DRoiPooling.cpp179 bool useNchw = context->getInputValue<bool>(kLayoutScalar); in prepare()
194 auto outputHeight = context->getInputValue<int32_t>(kOutputHeightScalar); in prepare()
195 auto outputWidth = context->getInputValue<int32_t>(kOutputWidthScalar); in prepare()
198 heightStride = context->getInputValue<_Float16>(kHeightStrideSalar); in prepare()
199 widthStride = context->getInputValue<_Float16>(kWidthStrideScalar); in prepare()
201 heightStride = context->getInputValue<float>(kHeightStrideSalar); in prepare()
202 widthStride = context->getInputValue<float>(kWidthStrideScalar); in prepare()
234 context->getInputValue<_Float16>(kHeightStrideSalar), in execute()
235 context->getInputValue<_Float16>(kWidthStrideScalar), in execute()
236 context->getInputValue<bool>(kLayoutScalar), in execute()
[all …]
DPooling.cpp59 padding_left = context->getInputValue<int32_t>(1); in initialize()
60 padding_right = context->getInputValue<int32_t>(2); in initialize()
61 padding_top = context->getInputValue<int32_t>(3); in initialize()
62 padding_bottom = context->getInputValue<int32_t>(4); in initialize()
63 stride_width = context->getInputValue<int32_t>(5); in initialize()
64 stride_height = context->getInputValue<int32_t>(6); in initialize()
65 filter_width = context->getInputValue<int32_t>(7); in initialize()
66 filter_height = context->getInputValue<int32_t>(8); in initialize()
67 activation = context->getInputValue<int32_t>(9); in initialize()
69 useNchw = context->getInputValue<bool>(10); in initialize()
[all …]
DInstanceNormalization.cpp107 context->getInputValue<_Float16>(kGammaScalar), in execute()
108 context->getInputValue<_Float16>(kBetaScalar), in execute()
109 context->getInputValue<_Float16>(kEpsilonScalar), in execute()
110 context->getInputValue<bool>(kLayoutScalar), in execute()
116 context->getInputValue<float>(kGammaScalar), in execute()
117 context->getInputValue<float>(kBetaScalar), in execute()
118 context->getInputValue<float>(kEpsilonScalar), in execute()
119 context->getInputValue<bool>(kLayoutScalar), in execute()
DBatchMatmul.cpp214 const bool adjX = context->getInputValue<bool>(kInputLHSAdj); in prepare()
215 const bool adjY = context->getInputValue<bool>(kInputRHSAdj); in prepare()
236 context->getInputValue<bool>(kInputLHSAdj), in execute()
237 context->getInputValue<bool>(kInputRHSAdj), in execute()
245 context->getInputValue<bool>(kInputLHSAdj), in execute()
246 context->getInputValue<bool>(kInputRHSAdj), in execute()
254 context->getInputValue<bool>(kInputLHSAdj), in execute()
255 context->getInputValue<bool>(kInputRHSAdj), in execute()
263 context->getInputValue<bool>(kInputLHSAdj), in execute()
264 context->getInputValue<bool>(kInputRHSAdj), in execute()
DLocalResponseNormalization.cpp120 ? context->getInputValue<int32_t>(kAxisScalar) in executeTyped()
125 context->getInputValue<int32_t>(kRadiusScalar), context->getInputValue<T>(kBiasScalar), in executeTyped()
126 context->getInputValue<T>(kAlphaScalar), context->getInputValue<T>(kBetaScalar), axis, in executeTyped()
136 ? context->getInputValue<int32_t>(kAxisScalar) in prepare()
141 const int32_t radius = context->getInputValue<int32_t>(kRadiusScalar); in prepare()
DDepthwiseConv2D.cpp59 padding_implicit = context->getInputValue<int32_t>(3); in initialize()
60 stride_width = context->getInputValue<int32_t>(4); in initialize()
61 stride_height = context->getInputValue<int32_t>(5); in initialize()
62 depth_multiplier = context->getInputValue<int32_t>(6); in initialize()
63 activation = context->getInputValue<int32_t>(7); in initialize()
65 useNchw = context->getInputValue<bool>(8); in initialize()
68 dilation_width_factor = context->getInputValue<int32_t>(9); in initialize()
69 dilation_height_factor = context->getInputValue<int32_t>(10); in initialize()
73 padding_left = context->getInputValue<int32_t>(3); in initialize()
74 padding_right = context->getInputValue<int32_t>(4); in initialize()
[all …]
DBroadcast.cpp460 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
468 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
476 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
484 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
492 context->getInputValue<int32_t>(kActivationScalar), in executeAdd()
510 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
518 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
526 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
534 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
542 context->getInputValue<int32_t>(kActivationScalar), in executeMul()
[all …]
DStridedSlice.cpp90 context->getInputValue<int32_t>(kBeginMask), context->getInputValue<int32_t>(kEndMask), in executeTyped()
91 context->getInputValue<int32_t>(kShrinkAxisMask), in executeTyped()
123 const int32_t beginMask = context->getInputValue<int32_t>(kBeginMask); in prepare()
124 const int32_t endMask = context->getInputValue<int32_t>(kEndMask); in prepare()
125 const int32_t shrinkAxisMask = context->getInputValue<int32_t>(kShrinkAxisMask); in prepare()
DChannelShuffle.cpp54 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in prepare()
55 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in prepare()
63 int32_t numGroups = context->getInputValue<int32_t>(kNumGroups); in execute()
64 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
DResizeImageOps.cpp162 scalarValue = context->getInputValue<bool>(scalarIndex); in getOptionalScalar()
191 height = context->getInputValue<int32_t>(kOutputHeightParamScalar); in prepare()
192 width = context->getInputValue<int32_t>(kOutputWidthParamScalar); in prepare()
195 context->getInputValue<float>(kOutputHeightParamScalar)); in prepare()
197 context->getInputValue<float>(kOutputWidthParamScalar)); in prepare()
201 static_cast<float>(context->getInputValue<_Float16>(kOutputHeightParamScalar))); in prepare()
204 static_cast<float>(context->getInputValue<_Float16>(kOutputWidthParamScalar))); in prepare()
DSoftmax.cpp234 ? context->getInputValue<_Float16>(kBetaScalar) in prepare()
235 : context->getInputValue<float>(kBetaScalar); in prepare()
247 ? context->getInputValue<int32_t>(kAxisScalar) in execute()
253 context->getInputValue<_Float16>(kBetaScalar), axis, in execute()
259 context->getInputValue<float>(kBetaScalar), axis, in execute()
265 context->getInputValue<float>(kBetaScalar), axis, in execute()
271 context->getInputValue<float>(kBetaScalar), axis, in execute()
DConv2D.cpp74 padding_implicit = context->getInputValue<int32_t>(3); in initialize()
75 stride_width = context->getInputValue<int32_t>(4); in initialize()
76 stride_height = context->getInputValue<int32_t>(5); in initialize()
77 activation = context->getInputValue<int32_t>(6); in initialize()
79 useNchw = context->getInputValue<bool>(7); in initialize()
82 dilation_width_factor = context->getInputValue<int32_t>(8); in initialize()
83 dilation_height_factor = context->getInputValue<int32_t>(9); in initialize()
87 padding_left = context->getInputValue<int32_t>(3); in initialize()
88 padding_right = context->getInputValue<int32_t>(4); in initialize()
89 padding_top = context->getInputValue<int32_t>(5); in initialize()
[all …]
DTransposeConv2D.cpp65 paddingImplicit = context->getInputValue<int32_t>(4); in initialize()
66 strideWidth = context->getInputValue<int32_t>(5); in initialize()
67 strideHeight = context->getInputValue<int32_t>(6); in initialize()
68 activation = context->getInputValue<int32_t>(7); in initialize()
69 useNchw = context->getInputValue<bool>(8); in initialize()
83 paddingLeft = context->getInputValue<int32_t>(3); in initialize()
84 paddingRight = context->getInputValue<int32_t>(4); in initialize()
85 paddingTop = context->getInputValue<int32_t>(5); in initialize()
86 paddingBottom = context->getInputValue<int32_t>(6); in initialize()
87 strideWidth = context->getInputValue<int32_t>(7); in initialize()
[all …]
DLogSoftmax.cpp70 int32_t axis = context->getInputValue<int32_t>(kInputAxis); in execute()
76 context->getInputValue<_Float16>(kInputBeta), axis, in execute()
81 context->getInputValue<float>(kInputBeta), axis, in execute()
DFullyConnected.cpp202 context->getInputValue<int32_t>(kActivationScalar), in execute()
212 context->getInputValue<int32_t>(kActivationScalar), in execute()
222 context->getInputValue<int32_t>(kActivationScalar), in execute()
232 context->getInputValue<int32_t>(kActivationScalar), in execute()
DElu.cpp61 context->getInputValue<_Float16>(kAlphaScalar), in execute()
67 context->getInputValue<float>(kAlphaScalar), in execute()
DQLSTM.cpp281 const float cellClip = context->getInputValue<float>(kCellClip); in execute()
282 const float projectionClip = context->getInputValue<float>(kProjectionClip); in execute()
283 const float inputIntermediateScale = context->getInputValue<float>(kInputIntermediateScale); in execute()
284 const float forgetIntermediateScale = context->getInputValue<float>(kForgetIntermediateScale); in execute()
285 const float cellIntermediateScale = context->getInputValue<float>(kCellIntermediateScale); in execute()
286 const float outputIntermediateScale = context->getInputValue<float>(kOutputIntermediateScale); in execute()
287 const int8_t hiddenStateZeroPoint = context->getInputValue<int8_t>(kHiddenStateZeroPoint); in execute()
288 const float hiddenStateScale = context->getInputValue<float>(kHiddenStateScale); in execute()
DUnidirectionalSequenceRNN.cpp62 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
67 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in executeTyped()
125 int32_t timeMajor = context->getInputValue<int32_t>(kTimeMajorParam); in prepare()
DConcatenation.cpp108 return concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
135 NN_RET_CHECK(concatenation(inputDatas, inputShapes, context->getInputValue<int32_t>(inputCount), in concatenation()
150 int32_t axis = context->getInputValue<int32_t>(numInputs - 1); in prepare()
DTopK_V2.cpp62 context->getInputValue<int32_t>(kTopKScalar), in executeTyped()
71 const int32_t k = context->getInputValue<int32_t>(kTopKScalar); in prepare()
DBidirectionalSequenceRNN.cpp129 const int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
130 const bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in executeTyped()
131 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in executeTyped()
289 const bool mergeOutputs = context->getInputValue<bool>(kMergeOutputsParam); in prepare()
325 bool timeMajor = context->getInputValue<bool>(kTimeMajorParam); in prepare()
DPack.cpp56 const int32_t axisDimension = context->getInputValue<int32_t>(kInputAxisScalar); in prepare()
83 const int32_t axis = context->getInputValue<int32_t>(kInputAxisScalar); in packParams()
/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