| /packages/modules/NeuralNetworks/common/operations/ |
| D | Activation.cpp | 44 constexpr uint32_t kInputTensor = 0; variable
365 auto inputType = context->getInputType(kInputTensor); in validate()
382 const Shape& input = context->getInputShape(kInputTensor); in validate()
394 auto inputType = context->getInputType(kInputTensor); in validateHardSwish()
410 Shape input = context->getInputShape(kInputTensor); in prepare()
446 switch (context->getInputType(kInputTensor)) { in executeRelu()
448 return reluFloat(context->getInputBuffer<_Float16>(kInputTensor), in executeRelu()
449 context->getInputShape(kInputTensor), in executeRelu()
453 return reluFloat(context->getInputBuffer<float>(kInputTensor), in executeRelu()
454 context->getInputShape(kInputTensor), in executeRelu()
[all …]
|
| D | Elementwise.cpp | 30 constexpr uint32_t kInputTensor = 0; variable
48 switch (context->getInputType(kInputTensor)) { in execute()
50 return compute<float, _Float16>(func, context->getInputBuffer<_Float16>(kInputTensor), in execute()
51 context->getInputShape(kInputTensor), in execute()
54 return compute<float, float>(func, context->getInputBuffer<float>(kInputTensor), in execute()
55 context->getInputShape(kInputTensor), in execute()
65 switch (context->getInputType(kInputTensor)) { in executeAbs()
68 context->getInputBuffer<_Float16>(kInputTensor), in executeAbs()
69 context->getInputShape(kInputTensor), in executeAbs()
72 return compute<float, float>(std::abs, context->getInputBuffer<float>(kInputTensor), in executeAbs()
[all …]
|
| D | ChannelShuffle.cpp | 30 constexpr uint32_t kInputTensor = 0; variable
63 auto inputType = context->getInputType(kInputTensor); in validate()
69 const Shape& inputShape = context->getInputShape(kInputTensor); in validate()
83 Shape input = context->getInputShape(kInputTensor); in prepare()
95 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
96 switch (context->getInputType(kInputTensor)) { in execute()
98 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
99 context->getInputShape(kInputTensor), numGroups, axis, in execute()
102 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
103 context->getInputShape(kInputTensor), numGroups, axis, in execute()
[all …]
|
| D | Gather.cpp | 30 constexpr uint32_t kInputTensor = 0; variable
65 OperandType inputType = context->getInputType(kInputTensor); in validate()
83 Shape input = context->getInputShape(kInputTensor); in prepare()
103 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
104 switch (context->getInputType(kInputTensor)) { in execute()
106 return eval(context->getInputBuffer<_Float16>(kInputTensor), in execute()
107 context->getInputShape(kInputTensor), axis, in execute()
112 return eval(context->getInputBuffer<float>(kInputTensor), in execute()
113 context->getInputShape(kInputTensor), axis, in execute()
118 return eval(context->getInputBuffer<int32_t>(kInputTensor), in execute()
[all …]
|
| D | Transpose.cpp | 38 constexpr uint32_t kInputTensor = 0; variable
81 const OperandType inputType = context->getInputType(kInputTensor); in validate()
92 const Shape& input = context->getInputShape(kInputTensor); in validate()
104 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
107 const Shape& input = context->getInputShape(kInputTensor); in prepare()
144 switch (context->getInputType(kInputTensor)) { in execute()
146 return transposeGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
147 context->getInputShape(kInputTensor), in execute()
153 return transposeGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
154 context->getInputShape(kInputTensor), in execute()
[all …]
|
| D | Neg.cpp | 32 constexpr uint32_t kInputTensor = 0; variable
53 OperandType inputType = context->getInputType(kInputTensor); in validate()
63 Shape input = context->getInputShape(kInputTensor); in prepare()
70 switch (context->getInputType(kInputTensor)) { in execute()
72 return compute(context->getInputBuffer<_Float16>(kInputTensor), in execute()
73 context->getInputShape(kInputTensor), in execute()
76 return compute(context->getInputBuffer<float>(kInputTensor), in execute()
77 context->getInputShape(kInputTensor), in execute()
80 return compute(context->getInputBuffer<int32_t>(kInputTensor), in execute()
81 context->getInputShape(kInputTensor), in execute()
|
| D | Reduce.cpp | 36 constexpr uint32_t kInputTensor = 0; variable
53 const Shape inputShape = context->getInputShape(kInputTensor); in compute()
61 context->getInputBuffer<T>(kInputTensor), in compute()
76 OperandType inputType = context->getInputType(kInputTensor); in validateProdSum()
83 const Shape& input = context->getInputShape(kInputTensor); in validateProdSum()
93 OperandType inputType = context->getInputType(kInputTensor); in validateMaxMin()
106 const Shape& input = context->getInputShape(kInputTensor); in validateMaxMin()
116 OperandType inputType = context->getInputType(kInputTensor); in validateLogical()
122 const Shape& input = context->getInputShape(kInputTensor); in validateLogical()
131 Shape inputShape = context->getInputShape(kInputTensor); in prepare()
[all …]
|
| D | Slice.cpp | 35 constexpr uint32_t kInputTensor = 0; variable
90 const OperandType inputType = context->getInputType(kInputTensor); in validate()
111 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
145 switch (context->getInputType(kInputTensor)) { in execute()
147 return evalGeneric(context->getInputBuffer<_Float16>(kInputTensor), in execute()
148 context->getInputShape(kInputTensor), in execute()
156 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
165 return evalGeneric(context->getInputBuffer<int32_t>(kInputTensor), in execute()
166 context->getInputShape(kInputTensor), in execute()
[all …]
|
| D | PRelu.cpp | 38 constexpr uint32_t kInputTensor = 0; variable
105 auto inputType = context->getInputType(kInputTensor); in validate()
122 Shape input = context->getInputShape(kInputTensor); in prepare()
131 switch (context->getInputType(kInputTensor)) { in execute()
137 context->getInputBuffer<_Float16>(kInputTensor), in execute()
138 context->getInputShape(kInputTensor), in execute()
148 context->getInputBuffer<float>(kInputTensor), in execute()
149 context->getInputShape(kInputTensor), in execute()
155 return evalQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
156 context->getInputShape(kInputTensor), in execute()
[all …]
|
| D | L2Normalization.cpp | 39 constexpr uint32_t kInputTensor = 0; variable
209 const OperandType inputType = context->getInputType(kInputTensor); in validate()
224 } else if (context->getInputShape(kInputTensor).dimensions.size() != 4) { in validate()
227 const Shape& input = context->getInputShape(kInputTensor); in validate()
238 const Shape& input = context->getInputShape(kInputTensor); in prepare()
266 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
267 switch (context->getInputType(kInputTensor)) { in execute()
269 return l2normFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
270 context->getInputShape(kInputTensor), axis, in execute()
274 return l2normFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
[all …]
|
| D | Dequantize.cpp | 28 constexpr uint32_t kInputTensor = 0; variable
82 const OperandType inputType = context->getInputType(kInputTensor); in validate()
85 const Shape& input = context->getInputShape(kInputTensor); in validate()
107 const Shape& input = context->getInputShape(kInputTensor); in prepare()
118 const OperandType inputType = context->getInputType(kInputTensor); in execute()
121 const Shape& inputShape = context->getInputShape(kInputTensor); in execute()
123 const uint8_t* inputBuffer = context->getInputBuffer<uint8_t>(kInputTensor); in execute()
131 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
139 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
147 const int8_t* inputBuffer = context->getInputBuffer<int8_t>(kInputTensor); in execute()
|
| D | Elu.cpp | 33 constexpr uint32_t kInputTensor = 0; variable
58 auto inputType = context->getInputType(kInputTensor); in validate()
73 Shape inputShape = context->getInputShape(kInputTensor); in prepare()
80 switch (context->getInputType(kInputTensor)) { in execute()
82 return eluFloat(context->getInputBuffer<_Float16>(kInputTensor), in execute()
83 context->getInputShape(kInputTensor), in execute()
88 return eluFloat(context->getInputBuffer<float>(kInputTensor), in execute()
89 context->getInputShape(kInputTensor), in execute()
|
| D | LogSoftmax.cpp | 34 constexpr uint32_t kInputTensor = 0; variable
76 OperandType inputType = context->getInputType(kInputTensor); in validate()
94 return context->setOutputShape(kOutputTensor, context->getInputShape(kInputTensor)); in prepare()
99 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
100 switch (context->getInputType(kInputTensor)) { in execute()
102 return compute(context->getInputBuffer<_Float16>(kInputTensor), in execute()
103 context->getInputShape(kInputTensor), in execute()
107 return compute(context->getInputBuffer<float>(kInputTensor), in execute()
108 context->getInputShape(kInputTensor), in execute()
|
| D | FullyConnected.cpp | 40 constexpr uint32_t kInputTensor = 0; variable
228 auto inputType = context->getInputType(kInputTensor); in validate()
252 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
284 Shape input = context->getInputShape(kInputTensor); in validate()
296 Shape input = context->getInputShape(kInputTensor); in prepare()
307 switch (context->getInputType(kInputTensor)) { in execute()
309 return fullyConnectedFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
310 context->getInputShape(kInputTensor), in execute()
319 return fullyConnectedFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
320 context->getInputShape(kInputTensor), in execute()
[all …]
|
| D | Quantize.cpp | 32 constexpr uint32_t kInputTensor = 0; variable
70 const OperandType inputType = context->getInputType(kInputTensor); in validate()
87 const Shape& input = context->getInputShape(kInputTensor); in prepare()
97 const OperandType inputType = context->getInputType(kInputTensor); in execute()
101 return quantizeToQuant8<float>(context->getInputBuffer<float>(kInputTensor), in execute()
105 return quantizeToQuant8Signed<float>(context->getInputBuffer<float>(kInputTensor), in execute()
111 return quantizeToQuant8<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
115 return quantizeToQuant8Signed<_Float16>(context->getInputBuffer<_Float16>(kInputTensor), in execute()
|
| D | Softmax.cpp | 43 constexpr uint32_t kInputTensor = 0; variable
239 auto inputType = context->getInputType(kInputTensor); in validate()
254 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
273 Shape input = context->getInputShape(kInputTensor); in prepare()
290 switch (context->getInputType(kInputTensor)) { in execute()
292 return softmaxFloat16(context->getInputBuffer<_Float16>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
298 return softmaxFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
299 context->getInputShape(kInputTensor), in execute()
304 return softmaxQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
|
| D | Squeeze.cpp | 32 constexpr uint32_t kInputTensor = 0; variable
41 OperandType inputType = context->getInputType(kInputTensor); in validate()
62 const Shape& input = context->getInputShape(kInputTensor); in validate()
72 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
76 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
128 switch (context->getInputType(kInputTensor)) { in execute()
133 return copyData(context->getInputBuffer(kInputTensor), in execute()
134 context->getInputShape(kInputTensor), in execute()
|
| D | RoiPooling.cpp | 39 constexpr uint32_t kInputTensor = 0; variable
196 auto inputType = context->getInputType(kInputTensor); in validate()
232 Shape input = context->getInputShape(kInputTensor); in prepare()
249 if (context->getInputType(kInputTensor) == OperandType::TENSOR_FLOAT16) { in prepare()
278 switch (context->getInputType(kInputTensor)) { in execute()
280 return roiPooling(context->getInputBuffer<_Float16>(kInputTensor), in execute()
281 context->getInputShape(kInputTensor), in execute()
292 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
304 return roiPooling(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
|
| D | InstanceNormalization.cpp | 36 constexpr uint32_t kInputTensor = 0; variable
111 auto inputType = context->getInputType(kInputTensor); in validate()
128 Shape input = context->getInputShape(kInputTensor); in prepare()
134 switch (context->getInputType(kInputTensor)) { in execute()
136 return instanceNorm(context->getInputBuffer<_Float16>(kInputTensor), in execute()
137 context->getInputShape(kInputTensor), in execute()
145 return instanceNorm(context->getInputBuffer<float>(kInputTensor), in execute()
146 context->getInputShape(kInputTensor), in execute()
|
| D | ResizeImageOps.cpp | 39 constexpr uint32_t kInputTensor = 0; variable
187 auto inputType = context->getInputType(kInputTensor); in validate()
231 Shape input = context->getInputShape(kInputTensor); in prepare()
289 switch (context->getInputType(kInputTensor)) { in execute()
291 return resizeImageOp(opType, context->getInputBuffer<_Float16>(kInputTensor), in execute()
292 context->getInputShape(kInputTensor), useNchw, alignCorners, in execute()
297 return resizeImageOp(opType, context->getInputBuffer<float>(kInputTensor), in execute()
298 context->getInputShape(kInputTensor), useNchw, alignCorners, in execute()
302 return resizeImageOp(opType, context->getInputBuffer<uint8_t>(kInputTensor), in execute()
303 context->getInputShape(kInputTensor), useNchw, alignCorners, in execute()
[all …]
|
| D | LocalResponseNormalization.cpp | 38 constexpr uint32_t kInputTensor = 0; variable
128 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in executeTyped()
130 context->getInputBuffer<T>(kInputTensor), context->getInputShape(kInputTensor), in executeTyped()
144 const OperandType inputType = context->getInputType(kInputTensor); in validate()
169 } else if (context->getInputShape(kInputTensor).dimensions.size() != 4) { in validate()
173 const Shape& input = context->getInputShape(kInputTensor); in validate()
184 const Shape& input = context->getInputShape(kInputTensor); in prepare()
198 switch (context->getInputType(kInputTensor)) { in execute()
|
| D | LogicalNot.cpp | 27 constexpr uint32_t kInputTensor = 0; variable
47 OperandType inputType = context->getInputType(kInputTensor); in validate()
56 Shape input = context->getInputShape(kInputTensor); in prepare()
63 return compute(context->getInputBuffer<bool8>(kInputTensor), in execute()
64 context->getInputShape(kInputTensor), in execute()
|
| D | DepthwiseConv2D.cpp | 39 constexpr uint32_t kInputTensor = 0; variable
96 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
426 auto inputType = context->getInputType(kInputTensor); in validate()
472 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
520 Shape input = context->getInputShape(kInputTensor); in prepare()
583 switch (context->getInputType(kInputTensor)) { in execute()
585 return depthwiseConv(context->getInputBuffer<float>(kInputTensor), in execute()
586 context->getInputShape(kInputTensor), in execute()
598 return depthwiseConv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
599 context->getInputShape(kInputTensor), in execute()
[all …]
|
| D | TransposeConv2D.cpp | 40 constexpr uint32_t kInputTensor = 0; variable
445 const auto inputType = context->getInputType(kInputTensor); in validate()
489 Shape input = context->getInputShape(kInputTensor); in prepare()
551 switch (context->getInputType(kInputTensor)) { in execute()
553 return transposeConv(context->getInputBuffer<float>(kInputTensor), in execute()
554 context->getInputShape(kInputTensor), in execute()
562 return transposeConv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
563 context->getInputShape(kInputTensor), in execute()
574 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
575 context->getInputShape(kInputTensor), in execute()
[all …]
|
| D | Conv2D.cpp | 45 constexpr uint32_t kInputTensor = 0; variable
107 Shape inputShape = context->getInputShape(kInputTensor); in initialize()
539 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
548 auto inputType = context->getInputType(kInputTensor); in validate()
589 const float inputScale = context->getInputShape(kInputTensor).scale; in validate()
637 Shape input = context->getInputShape(kInputTensor); in prepare()
704 switch (context->getInputType(kInputTensor)) { in execute()
706 return conv(context->getInputBuffer<float>(kInputTensor), in execute()
707 context->getInputShape(kInputTensor), in execute()
718 return conv(context->getInputBuffer<_Float16>(kInputTensor), in execute()
[all …]
|