| /frameworks/ml/nn/common/operations/ |
| D | Activation.cpp | 40 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, const Shape& outputShape, in reluFloat()
58 const Shape& outputShape) { in relu1Float()
68 const Shape& outputShape) { in relu6Float()
77 const Shape& outputShape) { in tanhFloat16()
87 const Shape& outputShape) { in tanhFloat32()
98 const Shape& outputShape) { in logisticFloat()
125 const Shape& outputShape) { in reluQuant8()
132 const Shape& outputShape) { in relu1Quant8()
139 const Shape& outputShape) { in relu6Quant8()
148 const Shape& outputShape) { in tanhQuant8()
[all …]
|
| D | Reshape.cpp | 33 const Shape& outputShape) { 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()
211 const Shape& outputShape) { 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()
224 const Shape& outputShape) { in l2Pool()
[all …]
|
| D | L2Normalization.cpp | 40 float* outputData, const Shape& outputShape) { in l2normFloat32Impl()
67 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8Impl()
99 const Shape& outputShape) { in l2normFloat32()
115 _Float16* outputData, const Shape& outputShape) { in l2normFloat16()
128 uint8_t* outputData, const Shape& outputShape) { in l2normQuant8()
|
| D | MaximumMinimum.cpp | 32 bool isMinimum, T* outputData, const Shape& outputShape) { in evalGeneric()
58 const Shape& outputShape) { in evalQuant8()
92 bool isMinimum, void* output, const Shape& outputShape) { in eval()
|
| D | Normalization.cpp | 32 const Shape& outputShape) { in localResponseNormFloat32Impl()
61 _Float16* outputData, const Shape& outputShape) { in localResponseNormFloat16()
76 const Shape& outputShape) { 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 | 41 const Shape& outputShape) { in copyToTensor()
72 const Shape& outputShape) { in eval()
|
| D | Quantize.cpp | 40 const Shape& outputShape) { in quantizeFloat32ToQuant8()
52 const Shape& outputShape) { in quantizeFloat16ToQuant8()
|
| D | ResizeImageOps.cpp | 48 T* outputData, const Shape& outputShape) { in resizeImageOpNhwc()
77 const Shape& outputShape) { in resizeImageOpNhwc()
90 T* outputData, const Shape& outputShape) { in resizeImageOp()
|
| D | SimpleMath.cpp | 53 const Shape& outputShape) { in meanFloat16()
67 bool keepDims, T* outputData, const Shape& outputShape) { in meanGeneric()
|
| 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 | Tile.cpp | 66 const Shape& outputShape) { in tileImpl()
87 uint8_t* outputData, const Shape& outputShape) { in eval()
|
| D | ExpandDims.cpp | 41 const Shape& outputShape) { in eval()
|
| D | Slice.cpp | 48 T* outputData, const Shape& outputShape) { in evalGeneric()
109 Shape outputShape = context->getOutputShape(kOutputTensor); in prepare() local
|
| D | DepthwiseConv2D.cpp | 34 _Float16* outputData, const Shape& outputShape) { in depthwiseConvFloat16()
71 const Shape& outputShape) { in depthwiseConvFloat32()
104 uint8_t* outputData, const Shape& outputShape) { in depthwiseConvQuant8()
155 const Shape& outputShape) { in depthwiseConvQuant8PerChannel()
|
| D | Concatenation.cpp | 39 const Shape& outputShape) { 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 | 118 const Shape& outputShape) { in transposeConvNhwc()
174 const Shape& outputShape) { in transposeConvNhwc()
270 const Shape& outputShape) { in transposeConvNhwc()
293 const Shape& outputShape) { in transposeConv()
309 uint8_t* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
411 uint8_t* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
|
| D | PRelu.cpp | 42 const Shape& outputShape) { in eval()
64 const Shape& bShape, uint8_t* outputData, const Shape& outputShape) { in evalQuant8()
|
| D | GroupedConv2D.cpp | 47 const Shape& outputShape) { in groupedConvFloat32()
104 const Shape& outputShape) { in groupedConvQuant8()
183 const Shape& outputShape) { 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()
158 const Shape& outputShape) { in roiPooling()
|