| /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 | Split.cpp | 30 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
56 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
63 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
70 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
77 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8Signed()
84 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
|
| D | Pooling.cpp | 79 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
139 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
149 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
162 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
172 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
185 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
195 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
207 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
217 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
227 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
[all …]
|
| D | Reshape.cpp | 34 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
43 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in depthToSpaceGeneric()
64 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in spaceToDepthGeneric()
85 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T padValue, in padGeneric()
198 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in batchToSpaceGeneric()
223 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in spaceToBatchGeneric()
|
| D | L2Normalization.cpp | 47 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis, in l2normFloat32Impl()
75 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl()
107 inline bool l2normQuant8SignedImpl(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8SignedImpl()
138 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
154 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16()
167 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8()
183 bool l2normQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Signed()
|
| D | TopK_V2.cpp | 44 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t k, T* valuesData, in evalGeneric()
99 const Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
117 const Shape inputShape = context->getInputShape(kInputTensor); in execute() local
|
| 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()
122 const Shape& inputShape = context->getInputShape(kInputTensor); in execute() local
|
| D | SimpleMath.cpp | 33 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
48 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
|
| D | Elu.cpp | 45 bool eluFloat(const T* inputData, const Shape& inputShape, const T alpha, T* outputData, in eluFloat()
74 Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| D | ArgMinMax.cpp | 33 static void argMinMaxImpl(const In* inputData, const Shape& inputShape, int32_t axis, bool isArgMin, in argMinMaxImpl()
55 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32 axis, bool isArgMin, in argMinMaxGeneric()
|
| D | Softmax.cpp | 51 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
83 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
99 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
114 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8Impl()
201 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8()
|
| D | ChannelShuffle.cpp | 41 inline bool eval(const T* inputData, const Shape& inputShape, int32_t numGroups, int32_t axis, in eval()
72 const Shape& inputShape = context->getInputShape(kInputTensor); in validate() local
|
| D | RNN.cpp | 72 const Shape& inputShape = input->shape(); in Prepare() local
120 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep()
140 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep()
|
| D | FullyConnected.cpp | 55 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
85 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
106 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
147 bool fullyConnectedQuant8(const int8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
|
| D | LocalResponseNormalization.cpp | 50 inline bool localResponseNormFloat32Impl(const float* inputData, const Shape& inputShape, in localResponseNormFloat32Impl()
85 bool localResponseNorm<float>(const float* inputData, const Shape& inputShape, int32_t radius, in localResponseNorm()
106 bool localResponseNorm<_Float16>(const _Float16* inputData, const Shape& inputShape, int32_t radius, in localResponseNorm()
|
| D | ResizeImageOps.cpp | 66 bool resizeNearestNeighbor(const T* inputData, const Shape& inputShape, bool alignCorners, in resizeNearestNeighbor()
109 bool resizeImageOpNhwc(OperationType opType, const T* inputData, const Shape& inputShape, in resizeImageOpNhwc()
137 const Shape& inputShape, bool alignCorners, bool halfPixelCenters, in resizeImageOpNhwc()
150 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
|
| D | Tile.cpp | 71 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
92 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples, in eval()
|
| D | RoiPooling.cpp | 54 inline bool roiPoolingNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPoolingNhwc()
143 inline bool roiPooling(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPooling()
159 inline bool roiPooling<uint8_t, uint16_t>(const uint8_t* inputData, const Shape& inputShape, in roiPooling()
174 inline bool roiPooling<int8_t, uint16_t>(const int8_t* inputData, const Shape& inputShape, in roiPooling()
|
| D | ExpandDims.cpp | 40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| D | Slice.cpp | 52 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t* beginData, in evalGeneric()
106 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| D | Conv2D.cpp | 107 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
193 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc()
222 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc()
276 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc()
305 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc()
332 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv()
351 bool convQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
444 bool convQuant8PerChannelNhwc(const int8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
508 bool convQuant8PerChannel(const T* inputData, const Shape& inputShape, const int8_t* filterData, in convQuant8PerChannel()
|
| D | DepthwiseConv2D.cpp | 94 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
131 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc()
165 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc()
191 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc()
242 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc()
275 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc()
371 bool depthwiseConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in depthwiseConv()
392 bool depthwiseConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel()
|
| D | InstanceNormalization.cpp | 48 inline bool instanceNormNhwc(const T* inputData, const Shape& inputShape, T gamma, T beta, in instanceNormNhwc()
91 inline bool instanceNorm(const T* inputData, const Shape& inputShape, T gamma, T beta, T epsilon, in instanceNorm()
|
| D | TransposeConv2D.cpp | 125 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc()
182 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc()
276 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc()
300 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv()
316 bool transposeConvQuant8PerChannelNhwc(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannelNhwc()
419 bool transposeConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannel()
|
| D | StridedSlice.cpp | 52 bool compute(const T* inputData, const Shape& inputShape, const int32_t* beginData, in compute()
140 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|