/frameworks/ml/nn/common/operations/ |
D | Activation.cpp | 44 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 …]
|
D | Elementwise.cpp | 31 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 …]
|
D | ChannelShuffle.cpp | 33 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 …]
|
D | Neg.cpp | 33 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()
|
D | Gather.cpp | 31 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 …]
|
D | Transpose.cpp | 37 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 …]
|
D | Elu.cpp | 36 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()
|
D | Slice.cpp | 33 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 …]
|
D | PRelu.cpp | 39 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 …]
|
D | Reduce.cpp | 35 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 …]
|
D | LogSoftmax.cpp | 35 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()
|
D | Dequantize.cpp | 29 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()
|
D | Quantize.cpp | 33 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()
|
D | Squeeze.cpp | 33 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()
|
D | L2Normalization.cpp | 37 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 …]
|
D | LogicalNot.cpp | 28 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()
|
D | FullyConnected.cpp | 38 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 …]
|
D | InstanceNormalization.cpp | 34 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()
|
D | Softmax.cpp | 40 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 …]
|
D | LocalResponseNormalization.cpp | 36 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()
|
D | TopK_V2.cpp | 32 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()
|
D | RoiPooling.cpp | 37 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 …]
|
D | ResizeImageOps.cpp | 38 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 …]
|
D | StridedSlice.cpp | 36 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()
|
D | Rank.cpp | 29 constexpr uint32_t kInputTensor = 0; variable 37 hal::OperandType inputType = context->getInputType(kInputTensor); in validate() 60 getNumberOfDimensions(context->getInputShape(kInputTensor)); in execute()
|