| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Activation.cpp | 54 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, in reluFloat()
72 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
82 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float()
91 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16()
101 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32()
112 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat()
127 inline bool reluXQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluXQuant8()
143 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8()
149 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8()
155 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8()
[all …]
|
| D | Pooling.cpp | 144 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
154 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
167 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
177 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
190 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
200 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
212 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
222 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
232 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
244 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
[all …]
|
| D | Reshape.cpp | 40 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
50 T* outputData, const Shape& outputShape) { in depthToSpaceGeneric()
71 T* outputData, const Shape& outputShape) { in spaceToDepthGeneric()
92 T* outputData, const Shape& outputShape) { in padGeneric()
205 T* outputData, const Shape& outputShape) { in batchToSpaceGeneric()
230 const int32_t* padding, const Shape& paddingShape, T* outputData, in spaceToBatchGeneric()
|
| D | L2Normalization.cpp | 47 float* outputData, const Shape& /*outputShape*/) { in l2normFloat32Impl()
75 uint8_t* outputData, const Shape& /*outputShape*/) { in l2normQuant8Impl()
107 int8_t* outputData, const Shape& /*outputShape*/) { in l2normQuant8SignedImpl()
137 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
154 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
167 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
183 int8_t* outputData, const Shape& outputShape) { in l2normQuant8Signed()
|
| D | FullyConnected.cpp | 55 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
85 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
106 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
147 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
|
| D | Softmax.cpp | 51 int32_t axis, float* outputData, const Shape& /*outputShape*/) { in softmaxSlowFloat32()
83 float* outputData, const Shape& outputShape) { in softmaxFloat32()
99 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
115 T* outputData, const Shape& /*outputShape*/) { in softmaxQuant8Impl()
201 T* outputData, const Shape& outputShape) { in softmaxQuant8()
|
| D | Quantize.cpp | 36 bool quantizeToQuant8(const T* inputData, uint8_t* outputData, const Shape& outputShape) { in quantizeToQuant8()
49 bool quantizeToQuant8Signed(const T* inputData, int8_t* outputData, const Shape& outputShape) { in quantizeToQuant8Signed()
|
| D | SimpleMath.cpp | 40 const Shape& axisShape, bool keepDims, _Float16* outputData, in meanFloat16()
55 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
|
| D | ResizeImageOps.cpp | 62 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
105 bool alignCorners, bool halfPixelCenters, T* outputData, in resizeImageOpNhwc()
133 _Float16* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
146 bool alignCorners, bool halfPixelCenters, T* outputData, in resizeImageOp()
|
| D | ArgMinMax.cpp | 32 Out* outputData, const Shape& /*outputShape*/) { in argMinMaxImpl()
54 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
|
| D | Pow.cpp | 35 const Shape& exponentShape, T* outputData, const Shape& outputShape) { in evalGeneric()
70 const Shape& exponentShape, void* outputData, const Shape& outputShape) { in eval()
|
| D | Cast.cpp | 44 bool copyToTensor(const FromT* inputData, int numElements, uint8_t* outputData, in copyToTensor()
72 bool eval(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in eval()
|
| D | LocalResponseNormalization.cpp | 47 int32_t axis, float* outputData, in localResponseNormFloat32Impl()
81 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNorm()
104 _Float16* outputData, const Shape& outputShape) { in localResponseNorm()
|
| D | RoiPooling.cpp | 45 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
134 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
151 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
166 float widthStride, bool useNchw, int8_t* outputData, in roiPooling()
|
| D | Dequantize.cpp | 31 bool compute(const InputType* inputData, const Shape& inputShape, OutputType* outputData) { in compute()
44 bool computePerChannel(const int8_t* inputData, const Shape& inputShape, OutputType* outputData) { in computePerChannel()
|
| D | BatchMatmul.cpp | 85 void transposeRowsColumns(const T* inputData, const Shape& inputShape, T* outputData) { in transposeRowsColumns()
117 const Shape& inputRHSShape, const bool adjX, const bool adjY, T* outputData, in batchMatMulGeneric()
154 T* outputData, const Shape& outputShape) { in batchMatMulQuantized()
|
| D | InstanceNormalization.cpp | 40 T epsilon, T* outputData, const Shape& /*outputShape*/) { in instanceNormNhwc()
83 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
|
| D | Conv2D.cpp | 196 float* outputData, const Shape& outputShape) { in convNhwc()
225 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
279 int8_t* outputData, Shape outputShape) { in convNhwc()
308 _Float16* outputData, const Shape& outputShape) { in convNhwc()
334 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv()
355 int32_t dilationHeightFactor, int32_t activation, uint8_t* outputData, in convQuant8PerChannelNhwc()
448 int32_t dilationHeightFactor, int32_t activation, int8_t* outputData, in convQuant8PerChannelNhwc()
512 T* outputData, const Shape& outputShape) { in convQuant8PerChannel()
|
| D | Tile.cpp | 69 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
91 uint8_t* outputData, const Shape& outputShape) { in eval()
|
| D | DepthwiseConv2D.cpp | 134 int32_t depthMultiplier, int32_t activation, float* outputData, in depthwiseConvNhwc()
169 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
194 int32_t depthMultiplier, int32_t activation, uint8_t* outputData, in depthwiseConvNhwc()
245 int32_t depthMultiplier, int32_t activation, int8_t* outputData, in depthwiseConvNhwc()
279 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
374 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv()
398 T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannel()
|
| D | MaximumMinimum.cpp | 36 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
62 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8()
|
| D | TransposeConv2D.cpp | 124 const TransposeConv2dParam& param, float* outputData, in transposeConvNhwc()
181 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
276 const TransposeConv2dParam& param, _Float16* outputData, in transposeConvNhwc()
299 const TransposeConv2dParam& param, T_Input* outputData, in transposeConv()
317 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
420 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
|
| D | Concatenation.cpp | 53 const std::vector<Shape>& inputShapes, int32_t axis, T* outputData, in concatenation()
74 uint8_t* outputData, const Shape& outputShape) { in concatenation()
|
| D | PRelu.cpp | 45 const Shape& aShape, const T* bData, const Shape& bShape, T* outputData, in eval()
69 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()
|