| /frameworks/ml/nn/common/operations/ |
| D | Activation.cpp | 52 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
69 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
79 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float()
88 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16()
98 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32()
109 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat()
124 inline bool reluXQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluXQuant8()
140 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8()
146 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8()
152 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8()
[all …]
|
| D | Pooling.cpp | 140 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
150 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
163 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
173 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
186 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
196 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
208 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
218 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
228 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
240 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
[all …]
|
| D | Reshape.cpp | 34 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
44 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
65 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
86 T* outputData, const Shape& outputShape) { in padGeneric()
199 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric()
224 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric()
|
| D | L2Normalization.cpp | 48 float* outputData, const Shape& outputShape) { in l2normFloat32Impl()
76 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl()
108 int8_t* outputData, const Shape& outputShape) { in l2normQuant8SignedImpl()
138 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
155 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
168 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
184 int8_t* outputData, const Shape& outputShape) { in l2normQuant8Signed()
|
| D | SimpleMath.cpp | 34 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
49 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
|
| D | ArgMinMax.cpp | 34 Out* outputData, const Shape& outputShape) { in argMinMaxImpl()
56 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
|
| D | Pow.cpp | 36 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
71 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
|
| D | Cast.cpp | 47 bool copyToTensor(const FromT* inputData, int numElements, uint8_t* outputData, in copyToTensor()
75 bool eval(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in eval()
|
| D | Quantize.cpp | 43 bool quantizeToQuant8(const T* inputData, uint8_t* outputData, const Shape& outputShape) { in quantizeToQuant8()
55 bool quantizeToQuant8Signed(const T* inputData, int8_t* outputData, const Shape& outputShape) { in quantizeToQuant8Signed()
|
| D | Softmax.cpp | 52 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32()
84 float* outputData, const Shape& outputShape) { in softmaxFloat32()
100 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
116 T* outputData, const Shape& outputShape) { in softmaxQuant8Impl()
202 T* outputData, const Shape& outputShape) { in softmaxQuant8()
|
| D | FullyConnected.cpp | 58 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
88 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
109 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
150 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
|
| D | LocalResponseNormalization.cpp | 52 int32_t axis, float* outputData, in localResponseNormFloat32Impl()
86 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNorm()
108 _Float16* outputData, const Shape& outputShape) { in localResponseNorm()
|
| D | ResizeImageOps.cpp | 67 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
110 bool alignCorners, bool halfPixelCenters, T* outputData, in resizeImageOpNhwc()
138 _Float16* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
151 bool alignCorners, bool halfPixelCenters, T* outputData, in resizeImageOp()
|
| D | Dequantize.cpp | 39 bool compute(const InputType* inputData, const Shape& inputShape, OutputType* outputData) { in compute()
51 bool computePerChannel(const int8_t* inputData, const Shape& inputShape, OutputType* outputData) { in computePerChannel()
|
| D | Tile.cpp | 71 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
93 uint8_t* outputData, const Shape& outputShape) { in eval()
|
| D | RoiPooling.cpp | 57 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
146 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
163 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
178 float widthStride, bool useNchw, int8_t* outputData, in roiPooling()
|
| D | MaximumMinimum.cpp | 38 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
64 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8()
|
| D | ExpandDims.cpp | 40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| D | InstanceNormalization.cpp | 49 T epsilon, T* outputData, const Shape& outputShape) { in instanceNormNhwc()
92 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
|
| D | Conv2D.cpp | 198 float* outputData, const Shape& outputShape) { in convNhwc()
227 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
281 int8_t* outputData, Shape outputShape) { in convNhwc()
310 _Float16* outputData, const Shape& outputShape) { in convNhwc()
336 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv()
357 int32_t dilationHeightFactor, int32_t activation, uint8_t* outputData, in convQuant8PerChannelNhwc()
450 int32_t dilationHeightFactor, int32_t activation, int8_t* outputData, in convQuant8PerChannelNhwc()
514 T* outputData, const Shape& outputShape) { in convQuant8PerChannel()
|
| D | TransposeConv2D.cpp | 127 const TransposeConv2dParam& param, float* outputData, in transposeConvNhwc()
184 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
279 const TransposeConv2dParam& param, _Float16* outputData, in transposeConvNhwc()
302 const TransposeConv2dParam& param, T_Input* outputData, in transposeConv()
320 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
423 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
|
| D | PRelu.cpp | 47 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval()
71 T* outputData, const Shape& outputShape) { in evalQuant8()
|
| D | DepthwiseConv2D.cpp | 136 int32_t depthMultiplier, int32_t activation, float* outputData, in depthwiseConvNhwc()
171 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
196 int32_t depthMultiplier, int32_t activation, uint8_t* outputData, in depthwiseConvNhwc()
247 int32_t depthMultiplier, int32_t activation, int8_t* outputData, in depthwiseConvNhwc()
281 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
376 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv()
400 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel()
|
| D | GroupedConv2D.cpp | 50 int32_t numGroups, int32_t activation, float* outputData, in groupedConvFloat32()
108 int32_t numGroups, int32_t activation, T* outputData, in groupedConvQuant8()
206 int32_t activation, T* outputData, const Shape& outputShape) { in groupedConvQuant8PerChannel()
290 int32_t activation, _Float16* outputData, const Shape& outputShape) { in groupedConvFloat16()
|
| D | Concatenation.cpp | 49 const std::vector<Shape>& inputShapes, int32_t axis, T* outputData, in concatenation()
70 uint8_t* outputData, const Shape& outputShape) { in concatenation()
|