| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A 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()
475 switch (context->getInputType(kInputTensor)) { in executeRelu1()
504 switch (context->getInputType(kInputTensor)) { in executeRelu6()
533 switch (context->getInputType(kInputTensor)) { in executeLogistic()
562 switch (context->getInputType(kInputTensor)) { in executeTanh()
[all …]
|
| H A D | Elementwise.cpp | 30 constexpr uint32_t kInputTensor = 0; variable
48 switch (context->getInputType(kInputTensor)) { in execute()
55 context->getInputShape(kInputTensor), in execute()
65 switch (context->getInputType(kInputTensor)) { in executeAbs()
73 context->getInputShape(kInputTensor), in executeAbs()
88 OperandType inputType = context->getInputType(kInputTensor); in validate()
100 OperandType inputType = context->getInputType(kInputTensor); in validateAbs()
113 OperandType inputType = context->getInputType(kInputTensor); in validateFloor()
120 const Shape& input = context->getInputShape(kInputTensor); in validateFloor()
129 Shape input = context->getInputShape(kInputTensor); in prepare()
[all …]
|
| H A 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()
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()
106 return eval(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
|
| H A 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()
104 switch (context->getInputType(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()
119 context->getInputShape(kInputTensor), axis, in execute()
125 context->getInputShape(kInputTensor), axis, in execute()
130 return eval(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
|
| H A 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()
|
| H A 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()
154 context->getInputShape(kInputTensor), in execute()
161 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A 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()
148 context->getInputShape(kInputTensor), in execute()
156 return evalGeneric(context->getInputBuffer<float>(kInputTensor), in execute()
157 context->getInputShape(kInputTensor), in execute()
166 context->getInputShape(kInputTensor), in execute()
175 context->getInputShape(kInputTensor), in execute()
183 return evalGeneric(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
|
| H A 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()
156 context->getInputShape(kInputTensor), in execute()
163 return evalQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
|
| H A D | Reduce.cpp | 36 constexpr uint32_t kInputTensor = 0; variable
61 context->getInputBuffer<T>(kInputTensor), in compute()
83 const Shape& input = context->getInputShape(kInputTensor); in validateProdSum()
106 const Shape& input = context->getInputShape(kInputTensor); in validateMaxMin()
131 Shape inputShape = context->getInputShape(kInputTensor); in prepare()
169 switch (context->getInputType(kInputTensor)) { in executeProd()
188 switch (context->getInputType(kInputTensor)) { in executeSum()
199 switch (context->getInputType(kInputTensor)) { in executeMax()
218 switch (context->getInputType(kInputTensor)) { in executeMin()
237 switch (context->getInputType(kInputTensor)) { in executeAny()
[all …]
|
| H A 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()
|
| H A 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()
|
| H A 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()
|
| H A 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()
|
| H A 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()
|
| H A 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()
267 switch (context->getInputType(kInputTensor)) { in execute()
269 return l2normFloat32(context->getInputBuffer<float>(kInputTensor), in execute()
270 context->getInputShape(kInputTensor), axis, in execute()
275 context->getInputShape(kInputTensor), axis, in execute()
279 return l2normQuant8(context->getInputBuffer<uint8_t>(kInputTensor), in execute()
[all …]
|
| H A 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()
320 context->getInputShape(kInputTensor), in execute()
330 context->getInputShape(kInputTensor), in execute()
[all …]
|
| H A 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()
|
| H A 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()
|
| H A D | Softmax.cpp | 43 constexpr uint32_t kInputTensor = 0; variable
239 auto inputType = context->getInputType(kInputTensor); in validate()
273 Shape input = context->getInputShape(kInputTensor); in prepare()
290 switch (context->getInputType(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()
305 context->getInputShape(kInputTensor), in execute()
310 return softmaxQuant8(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
|
| H A 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()
|
| H A 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()
281 context->getInputShape(kInputTensor), in execute()
292 return roiPooling(context->getInputBuffer<float>(kInputTensor), in execute()
293 context->getInputShape(kInputTensor), in execute()
305 context->getInputShape(kInputTensor), in execute()
316 return roiPooling(context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
|
| H A 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 …]
|
| H A D | TopK_V2.cpp | 31 constexpr uint32_t kInputTensor = 0; variable
67 return evalGeneric(context->getInputBuffer<T>(kInputTensor), in executeTyped()
68 context->getInputShape(kInputTensor), in executeTyped()
79 OperandType inputType = context->getInputType(kInputTensor); in validate()
96 const Shape inputShape = context->getInputShape(kInputTensor); in prepare()
114 const Shape inputShape = context->getInputShape(kInputTensor); in execute()
|
| H A D | StridedSlice.cpp | 38 constexpr uint32_t kInputTensor = 0; variable
92 context->getInputBuffer<T>(kInputTensor), context->getInputShape(kInputTensor), in executeTyped()
107 OperandType inputType = context->getInputType(kInputTensor); in validate()
133 const Shape& input = context->getInputShape(kInputTensor); in validate()
143 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare()
209 switch (context->getInputType(kInputTensor)) { in execute()
|
| H A 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()
554 context->getInputShape(kInputTensor), in execute()
563 context->getInputShape(kInputTensor), in execute()
574 context->getInputBuffer<uint8_t>(kInputTensor), in execute()
575 context->getInputShape(kInputTensor), in execute()
587 context->getInputShape(kInputTensor), in execute()
601 context->getInputBuffer<int8_t>(kInputTensor), in execute()
[all …]
|