Home
last modified time | relevance | path

Searched refs:kInputTensor (Results 1 – 25 of 50) sorted by relevance

12

/frameworks/ml/nn/common/operations/
DActivation.cpp44 constexpr uint32_t kInputTensor = 0; variable
362 auto inputType = context->getInputType(kInputTensor); in validate()
378 const Shape& input = context->getInputShape(kInputTensor); in validate()
388 auto inputType = context->getInputType(kInputTensor); in validateHardSwish()
400 Shape input = context->getInputShape(kInputTensor); in prepare()
436 switch (context->getInputType(kInputTensor)) { in executeRelu()
438 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
439 context->getInputShape(kInputTensor), in executeRelu()
443 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
444 context->getInputShape(kInputTensor), in executeRelu()
[all …]
DElementwise.cpp31 constexpr uint32_t kInputTensor = 0; variable
51 switch (context->getInputType(kInputTensor)) { in execute()
53 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
54 context->getInputShape(kInputTensor), in execute()
57 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
58 context->getInputShape(kInputTensor), in execute()
68 switch (context->getInputType(kInputTensor)) { in executeAbs()
71 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
72 context->getInputShape(kInputTensor), in executeAbs()
75 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
[all …]
DChannelShuffle.cpp33 constexpr uint32_t kInputTensor = 0; variable
66 auto inputType = context->getInputType(kInputTensor); in validate()
72 const Shape& inputShape = context->getInputShape(kInputTensor); in validate()
86 Shape input = context->getInputShape(kInputTensor); in prepare()
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
99 switch (context->getInputType(kInputTensor)) { in execute()
101 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
102 context->getInputShape(kInputTensor), numGroups, axis, in execute()
105 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
106 context->getInputShape(kInputTensor), numGroups, axis, in execute()
[all …]
DNeg.cpp33 constexpr uint32_t kInputTensor = 0; variable
56 OperandType inputType = context->getInputType(kInputTensor); in validate()
66 Shape input = context->getInputShape(kInputTensor); in prepare()
73 switch (context->getInputType(kInputTensor)) { in execute()
75 return compute(context->getInputBuffer<_Float16>(kInputTensor), in execute()
76 context->getInputShape(kInputTensor), in execute()
79 return compute(context->getInputBuffer<float>(kInputTensor), in execute()
80 context->getInputShape(kInputTensor), in execute()
83 return compute(context->getInputBuffer<int32_t>(kInputTensor), in execute()
84 context->getInputShape(kInputTensor), in execute()
DGather.cpp31 constexpr uint32_t kInputTensor = 0; variable
68 OperandType inputType = context->getInputType(kInputTensor); in validate()
86 Shape input = context->getInputShape(kInputTensor); in prepare()
106 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
107 switch (context->getInputType(kInputTensor)) { in execute()
109 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
110 context->getInputShape(kInputTensor), axis, in execute()
115 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
116 context->getInputShape(kInputTensor), axis, in execute()
121 return eval(context->getInputBuffer<int32_t>(kInputTensor), in execute()
[all …]
DTranspose.cpp37 constexpr uint32_t kInputTensor = 0; variable
80 const OperandType inputType = context->getInputType(kInputTensor); in validate()
90 const Shape& input = context->getInputShape(kInputTensor); in validate()
100 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
103 const Shape& input = context->getInputShape(kInputTensor); in prepare()
140 switch (context->getInputType(kInputTensor)) { in execute()
142 return transposeGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
143 context->getInputShape(kInputTensor), in execute()
149 return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
150 context->getInputShape(kInputTensor), in execute()
[all …]
DElu.cpp36 constexpr uint32_t kInputTensor = 0; variable
61 auto inputType = context->getInputType(kInputTensor); in validate()
74 Shape inputShape = context->getInputShape(kInputTensor); in prepare()
81 switch (context->getInputType(kInputTensor)) { in execute()
83 return eluFloat(context->getInputBuffer<_Float16>(kInputTensor), in execute()
84 context->getInputShape(kInputTensor), in execute()
89 return eluFloat(context->getInputBuffer<float>(kInputTensor), in execute()
90 context->getInputShape(kInputTensor), in execute()
DSlice.cpp33 constexpr uint32_t kInputTensor = 0; variable
88 const OperandType inputType = context->getInputType(kInputTensor); in validate()
106 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
140 switch (context->getInputType(kInputTensor)) { in execute()
142 return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
143 context->getInputShape(kInputTensor), in execute()
151 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
152 context->getInputShape(kInputTensor), in execute()
160 return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor), in execute()
161 context->getInputShape(kInputTensor), in execute()
[all …]
DPRelu.cpp39 constexpr uint32_t kInputTensor = 0; variable
104 auto inputType = context->getInputType(kInputTensor); in validate()
120 Shape input = context->getInputShape(kInputTensor); in prepare()
129 switch (context->getInputType(kInputTensor)) { in execute()
135 context->getInputBuffer<_Float16>(kInputTensor), in execute()
136 context->getInputShape(kInputTensor), in execute()
146 context->getInputBuffer<float>(kInputTensor), in execute()
147 context->getInputShape(kInputTensor), in execute()
153 return evalQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
154 context->getInputShape(kInputTensor), in execute()
[all …]
DReduce.cpp35 constexpr uint32_t kInputTensor = 0; variable
53 const Shape inputShape = context->getInputShape(kInputTensor); in compute()
61 context->getInputBuffer<T>(kInputTensor), in compute()
75 OperandType inputType = context->getInputType(kInputTensor); in validateProdSum()
82 const Shape& input = context->getInputShape(kInputTensor); in validateProdSum()
92 OperandType inputType = context->getInputType(kInputTensor); in validateMaxMin()
105 const Shape& input = context->getInputShape(kInputTensor); in validateMaxMin()
115 OperandType inputType = context->getInputType(kInputTensor); in validateLogical()
121 const Shape& input = context->getInputShape(kInputTensor); in validateLogical()
129 Shape inputShape = context->getInputShape(kInputTensor); in prepare()
[all …]
DLogSoftmax.cpp35 constexpr uint32_t kInputTensor = 0; variable
77 OperandType inputType = context->getInputType(kInputTensor); in validate()
96 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
101 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
102 switch (context->getInputType(kInputTensor)) { in execute()
104 return compute(context->getInputBuffer<_Float16>(kInputTensor), in execute()
105 context->getInputShape(kInputTensor), in execute()
109 return compute(context->getInputBuffer<float>(kInputTensor), in execute()
110 context->getInputShape(kInputTensor), in execute()
DDequantize.cpp29 constexpr uint32_t kInputTensor = 0; variable
83 const OperandType inputType = context->getInputType(kInputTensor); in validate()
86 const Shape& input = context->getInputShape(kInputTensor); in validate()
108 const Shape& input = context->getInputShape(kInputTensor); in prepare()
119 const OperandType inputType = context->getInputType(kInputTensor); in execute()
122 const Shape& inputShape = context->getInputShape(kInputTensor); in execute()
124 const uint8_t* inputBuffer = context->getInputBuffer<uint8_t>(kInputTensor); in execute()
132 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
140 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
148 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
DQuantize.cpp33 constexpr uint32_t kInputTensor = 0; variable
73 const OperandType inputType = context->getInputType(kInputTensor); in validate()
90 const Shape& input = context->getInputShape(kInputTensor); in prepare()
100 const OperandType inputType = context->getInputType(kInputTensor); in execute()
104 return quantizeToQuant8<float>(context->getInputBuffer<float>(kInputTensor), in execute()
108 return quantizeToQuant8Signed<float>(context->getInputBuffer<float>(kInputTensor), in execute()
114 return quantizeToQuant8<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
118 return quantizeToQuant8Signed<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
DSqueeze.cpp33 constexpr uint32_t kInputTensor = 0; variable
44 OperandType inputType = context->getInputType(kInputTensor); in validate()
65 const Shape& input = context->getInputShape(kInputTensor); in validate()
74 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
78 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
130 switch (context->getInputType(kInputTensor)) { in execute()
135 return copyData(context->getInputBuffer(kInputTensor), in execute()
136 context->getInputShape(kInputTensor), in execute()
DL2Normalization.cpp37 constexpr uint32_t kInputTensor = 0; variable
207 const OperandType inputType = context->getInputType(kInputTensor); in validate()
221 } else if (context->getInputShape(kInputTensor).dimensions.size() != 4) { in validate()
224 const Shape& input = context->getInputShape(kInputTensor); in validate()
233 const Shape& input = context->getInputShape(kInputTensor); in prepare()
261 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
262 switch (context->getInputType(kInputTensor)) { in execute()
264 return l2normFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
265 context->getInputShape(kInputTensor), axis, in execute()
269 return l2normFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
[all …]
DLogicalNot.cpp28 constexpr uint32_t kInputTensor = 0; variable
50 OperandType inputType = context->getInputType(kInputTensor); in validate()
59 Shape input = context->getInputShape(kInputTensor); in prepare()
66 return compute(context->getInputBuffer<bool8>(kInputTensor), in execute()
67 context->getInputShape(kInputTensor), in execute()
DFullyConnected.cpp38 constexpr uint32_t kInputTensor = 0; variable
226 auto inputType = context->getInputType(kInputTensor); in validate()
249 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
282 Shape input = context->getInputShape(kInputTensor); in validate()
293 Shape input = context->getInputShape(kInputTensor); in prepare()
304 switch (context->getInputType(kInputTensor)) { in execute()
306 return fullyConnectedFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
307 context->getInputShape(kInputTensor), in execute()
316 return fullyConnectedFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
317 context->getInputShape(kInputTensor), in execute()
[all …]
DInstanceNormalization.cpp34 constexpr uint32_t kInputTensor = 0; variable
109 auto inputType = context->getInputType(kInputTensor); in validate()
126 Shape input = context->getInputShape(kInputTensor); in prepare()
132 switch (context->getInputType(kInputTensor)) { in execute()
134 return instanceNorm(context->getInputBuffer<_Float16>(kInputTensor), in execute()
135 context->getInputShape(kInputTensor), in execute()
143 return instanceNorm(context->getInputBuffer<float>(kInputTensor), in execute()
144 context->getInputShape(kInputTensor), in execute()
DSoftmax.cpp40 constexpr uint32_t kInputTensor = 0; variable
236 auto inputType = context->getInputType(kInputTensor); in validate()
250 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
267 Shape input = context->getInputShape(kInputTensor); in prepare()
284 switch (context->getInputType(kInputTensor)) { in execute()
286 return softmaxFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
287 context->getInputShape(kInputTensor), in execute()
292 return softmaxFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
298 return softmaxQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
DLocalResponseNormalization.cpp36 constexpr uint32_t kInputTensor = 0; variable
126 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in executeTyped()
128 context->getInputBuffer<T>(kInputTensor), context->getInputShape(kInputTensor), in executeTyped()
141 const OperandType inputType = context->getInputType(kInputTensor); in validate()
165 } else if (context->getInputShape(kInputTensor).dimensions.size() != 4) { in validate()
169 const Shape& input = context->getInputShape(kInputTensor); in validate()
178 const Shape& input = context->getInputShape(kInputTensor); in prepare()
190 switch (context->getInputType(kInputTensor)) { in execute()
DTopK_V2.cpp32 constexpr uint32_t kInputTensor = 0; variable
70 return evalGeneric(context->getInputBuffer<T>(kInputTensor), in executeTyped()
71 context->getInputShape(kInputTensor), in executeTyped()
82 OperandType inputType = context->getInputType(kInputTensor); in validate()
99 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
117 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
DRoiPooling.cpp37 constexpr uint32_t kInputTensor = 0; variable
194 auto inputType = context->getInputType(kInputTensor); in validate()
231 Shape input = context->getInputShape(kInputTensor); in prepare()
248 if (context->getInputType(kInputTensor) == OperandType::TENSOR_FLOAT16) { in prepare()
277 switch (context->getInputType(kInputTensor)) { in execute()
279 return roiPooling(context->getInputBuffer<_Float16>(kInputTensor), in execute()
280 context->getInputShape(kInputTensor), in execute()
291 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
292 context->getInputShape(kInputTensor), in execute()
303 return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
DResizeImageOps.cpp38 constexpr uint32_t kInputTensor = 0; variable
184 auto inputType = context->getInputType(kInputTensor); in validate()
225 Shape input = context->getInputShape(kInputTensor); in prepare()
283 switch (context->getInputType(kInputTensor)) { in execute()
285 return resizeImageOp(opType, context->getInputBuffer<_Float16>(kInputTensor), in execute()
286 context->getInputShape(kInputTensor), useNchw, alignCorners, in execute()
291 return resizeImageOp(opType, context->getInputBuffer<float>(kInputTensor), in execute()
292 context->getInputShape(kInputTensor), useNchw, alignCorners, in execute()
296 return resizeImageOp(opType, context->getInputBuffer<uint8_t>(kInputTensor), in execute()
297 context->getInputShape(kInputTensor), useNchw, alignCorners, in execute()
[all …]
DStridedSlice.cpp36 constexpr uint32_t kInputTensor = 0; variable
91 context->getInputBuffer<T>(kInputTensor), context->getInputShape(kInputTensor), in executeTyped()
105 OperandType inputType = context->getInputType(kInputTensor); in validate()
131 const Shape& input = context->getInputShape(kInputTensor); in validate()
140 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
206 switch (context->getInputType(kInputTensor)) { in execute()
DRank.cpp29 constexpr uint32_t kInputTensor = 0; variable
37 hal::OperandType inputType = context->getInputType(kInputTensor); in validate()
60 getNumberOfDimensions(context->getInputShape(kInputTensor)); in execute()

12