| /packages/modules/NeuralNetworks/common/operations/ |
| D | Activation.cpp | 53 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
70 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
80 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float()
89 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16()
99 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32()
110 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat()
125 inline bool reluXQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluXQuant8()
141 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8()
147 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8()
153 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8()
[all …]
|
| D | Pooling.cpp | 142 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
152 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
165 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
175 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
188 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
198 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
210 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
220 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
230 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
242 _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 | 49 float* outputData, const Shape& outputShape) { in l2normFloat32Impl()
77 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl()
109 int8_t* outputData, const Shape& outputShape) { in l2normQuant8SignedImpl()
139 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
156 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
169 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
185 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 | 30 Out* outputData, const Shape& outputShape) { in argMinMaxImpl()
52 uint8_t* outputData, const Shape& outputShape) { in argMinMaxGeneric()
|
| 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 | 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 | Quantize.cpp | 40 bool quantizeToQuant8(const T* inputData, uint8_t* outputData, const Shape& outputShape) { in quantizeToQuant8()
52 bool quantizeToQuant8Signed(const T* inputData, int8_t* outputData, const Shape& outputShape) { in quantizeToQuant8Signed()
|
| D | Softmax.cpp | 54 int32_t axis, float* outputData, const Shape& outputShape) { in softmaxSlowFloat32()
86 float* outputData, const Shape& outputShape) { in softmaxFloat32()
102 int32_t axis, _Float16* outputData, const Shape& outputShape) { in softmaxFloat16()
118 T* outputData, const Shape& outputShape) { in softmaxQuant8Impl()
204 T* outputData, const Shape& outputShape) { in softmaxQuant8()
|
| D | FullyConnected.cpp | 59 float* outputData, const Shape& outputShape) { in fullyConnectedFloat32()
89 _Float16* outputData, const Shape& outputShape) { in fullyConnectedFloat16()
110 uint8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
151 int8_t* outputData, const Shape& outputShape) { in fullyConnectedQuant8()
|
| D | ResizeImageOps.cpp | 69 bool halfPixelCenters, T* outputData, const Shape& outputShape) { in resizeNearestNeighbor()
112 bool alignCorners, bool halfPixelCenters, T* outputData, in resizeImageOpNhwc()
140 _Float16* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
153 bool alignCorners, bool halfPixelCenters, T* outputData, in resizeImageOp()
|
| D | ExpandDims.cpp | 40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| D | LocalResponseNormalization.cpp | 53 int32_t axis, float* outputData, in localResponseNormFloat32Impl()
87 float bias, float alpha, float beta, int32_t axis, float* outputData, in localResponseNorm()
110 _Float16* outputData, const Shape& outputShape) { in localResponseNorm()
|
| 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 | Dequantize.cpp | 36 bool compute(const InputType* inputData, const Shape& inputShape, OutputType* outputData) { in compute()
48 bool computePerChannel(const int8_t* inputData, const Shape& inputShape, OutputType* outputData) { in computePerChannel()
|
| D | MaximumMinimum.cpp | 36 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
62 bool isMinimum, T* outputData, const Shape& outputShape) { in evalQuant8()
|
| D | RoiPooling.cpp | 58 T_Input* outputData, const Shape& outputShape) { in roiPoolingNhwc()
147 bool useNchw, T_Input* outputData, const Shape& outputShape) { in roiPooling()
164 float widthStride, bool useNchw, uint8_t* outputData, in roiPooling()
179 float widthStride, bool useNchw, int8_t* outputData, in roiPooling()
|
| D | InstanceNormalization.cpp | 50 T epsilon, T* outputData, const Shape& outputShape) { in instanceNormNhwc()
93 bool useNchw, T* outputData, const Shape& outputShape) { in instanceNorm()
|
| D | Conv2D.cpp | 199 float* outputData, const Shape& outputShape) { in convNhwc()
228 uint8_t* outputData, const Shape& outputShape) { in convNhwc()
282 int8_t* outputData, Shape outputShape) { in convNhwc()
311 _Float16* outputData, const Shape& outputShape) { in convNhwc()
337 int32_t dilation_height_factor, int32_t activation, bool useNchw, T_Input* outputData, in conv()
358 int32_t dilationHeightFactor, int32_t activation, uint8_t* outputData, in convQuant8PerChannelNhwc()
451 int32_t dilationHeightFactor, int32_t activation, int8_t* outputData, in convQuant8PerChannelNhwc()
515 T* outputData, const Shape& outputShape) { in convQuant8PerChannel()
|
| 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 | TransposeConv2D.cpp | 128 const TransposeConv2dParam& param, float* outputData, in transposeConvNhwc()
185 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
280 const TransposeConv2dParam& param, _Float16* outputData, in transposeConvNhwc()
303 const TransposeConv2dParam& param, T_Input* outputData, in transposeConv()
321 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
424 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
|
| D | DepthwiseConv2D.cpp | 138 int32_t depthMultiplier, int32_t activation, float* outputData, in depthwiseConvNhwc()
173 _Float16* outputData, const Shape& outputShape) { in depthwiseConvNhwc()
198 int32_t depthMultiplier, int32_t activation, uint8_t* outputData, in depthwiseConvNhwc()
249 int32_t depthMultiplier, int32_t activation, int8_t* outputData, in depthwiseConvNhwc()
283 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
378 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv()
402 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 | RNN.cpp | 119 const int32_t activation, T* outputData) { in RNNStep()
141 const uint32_t outputBatchStride, const uint32_t outputBatchOffset, T* outputData, in RNNStep()
|