| /frameworks/ml/nn/common/operations/ |
| D | Activation.cpp | 40 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
57 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
67 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float()
76 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16()
86 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32()
97 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat()
124 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8()
131 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8()
138 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8()
147 bool tanhQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in tanhQuant8()
[all …]
|
| D | SimpleMath.cpp | 32 bool floorFloat16(const _Float16* inputData, _Float16* outputData, const Shape& shape) { in floorFloat16()
43 bool floorFloat32(const float* inputData, float* outputData, const Shape& shape) { in floorFloat32()
52 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
67 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
|
| D | Reshape.cpp | 32 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
42 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
60 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
78 T* outputData, const Shape& outputShape) { in padGeneric()
188 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric()
210 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric()
|
| D | Pooling.cpp | 122 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
132 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
145 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
155 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
165 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
177 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
187 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
197 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
211 T* outputData, const Shape& outputShape) { in averagePool()
223 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool()
[all …]
|
| D | L2Normalization.cpp | 40 float* outputData, const Shape& outputShape) { in l2normFloat32Impl()
67 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl()
98 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
115 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
128 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
|
| D | Normalization.cpp | 31 int32_t axis, float* outputData, in localResponseNormFloat32Impl()
61 _Float16* outputData, const Shape& outputShape) { in localResponseNormFloat16()
75 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNormFloat32()
|
| D | ArgMinMax.cpp | 32 Out* outputData, const Shape& outputShape) { in argMinMaxImpl()
57 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
|
| D | Softmax.cpp | 43 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32()
75 float* outputData, const Shape& outputShape) { in softmaxFloat32()
91 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
106 uint8_t* outputData, const Shape& outputShape) { in softmaxQuant8Impl()
187 int32_t axis, uint8_t* outputData, const Shape& outputShape) { in softmaxQuant8()
|
| D | Pow.cpp | 33 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
68 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
|
| D | Cast.cpp | 40 bool copyToTensor(const FromT* inputData, int numElements, uint8_t* outputData, in copyToTensor()
71 bool eval(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in eval()
|
| D | Quantize.cpp | 39 bool quantizeFloat32ToQuant8(const float* inputData, uint8_t* outputData, in quantizeFloat32ToQuant8()
51 bool quantizeFloat16ToQuant8(const _Float16* inputData, uint8_t* outputData, in quantizeFloat16ToQuant8()
|
| D | ResizeImageOps.cpp | 48 T* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
76 const Shape& inputShape, _Float16* outputData, in resizeImageOpNhwc()
90 T* outputData, const Shape& outputShape) { in resizeImageOp()
|
| D | FullyConnected.cpp | 50 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
80 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
101 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
|
| D | Dequantize.cpp | 37 bool compute(const InputType* inputData, const Shape& inputShape, OutputType* outputData) { in compute()
49 bool computePerChannel(const int8_t* inputData, const Shape& inputShape, OutputType* outputData) { in computePerChannel()
|
| D | Tile.cpp | 65 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
87 uint8_t* outputData, const Shape& outputShape) { in eval()
|
| D | MaximumMinimum.cpp | 32 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
57 const Shape& bShape, bool isMinimum, uint8_t* outputData, in evalQuant8()
|
| D | ExpandDims.cpp | 40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| D | DepthwiseConv2D.cpp | 34 _Float16* outputData, const Shape& outputShape) { in depthwiseConvFloat16()
70 int32_t depthMultiplier, int32_t activation, float* outputData, in depthwiseConvFloat32()
104 uint8_t* outputData, const Shape& outputShape) { in depthwiseConvQuant8()
154 int32_t depthMultiplier, int32_t activation, uint8_t* outputData, in depthwiseConvQuant8PerChannel()
|
| D | Concatenation.cpp | 38 const std::vector<Shape>& inputShapes, int32_t axis, T* outputData, in concatenation()
59 uint8_t* outputData, const Shape& outputShape) { in concatenation()
|
| D | InstanceNormalization.cpp | 47 T epsilon, T* outputData, const Shape& outputShape) { in instanceNormNhwc()
79 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
|
| D | TransposeConv2D.cpp | 117 const TransposeConv2dParam& param, float* outputData, in transposeConvNhwc()
173 const TransposeConv2dParam& param, uint8_t* outputData, in transposeConvNhwc()
269 const TransposeConv2dParam& param, _Float16* outputData, in transposeConvNhwc()
292 const TransposeConv2dParam& param, T_Input* outputData, in transposeConv()
309 uint8_t* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
411 uint8_t* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
|
| D | PRelu.cpp | 41 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval()
64 const Shape& bShape, uint8_t* outputData, const Shape& outputShape) { in evalQuant8()
|
| D | GroupedConv2D.cpp | 46 int32_t numGroups, int32_t activation, float* outputData, in groupedConvFloat32()
103 int32_t numGroups, int32_t activation, uint8_t* outputData, in groupedConvQuant8()
182 int32_t activation, uint8_t* outputData, in groupedConvQuant8PerChannel()
267 int32_t activation, _Float16* outputData, const Shape& outputShape) { in groupedConvFloat16()
|
| D | StridedSlice.cpp | 34 int32_t shrinkAxisMask, uint8_t* outputData, const Shape& outputShape) { in stridedSliceGeneric()
|
| D | RoiPooling.cpp | 51 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
140 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
157 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
|