| /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 | Split.cpp | 28 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
54 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
61 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
68 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
75 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
|
| D | Pooling.cpp | 71 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
121 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
131 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
144 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
154 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
164 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
176 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
186 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
196 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
210 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool()
[all …]
|
| D | Reshape.cpp | 32 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
41 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in depthToSpaceGeneric()
59 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in spaceToDepthGeneric()
77 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T padValue, in padGeneric()
187 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in batchToSpaceGeneric()
209 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in spaceToBatchGeneric()
|
| D | L2Normalization.cpp | 39 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis, in l2normFloat32Impl()
66 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl()
98 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
114 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16()
127 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8()
|
| 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()
113 const Shape& inputShape = context->getInputShape(kInputTensor); in execute() local
|
| D | Normalization.cpp | 29 inline bool localResponseNormFloat32Impl(const float* inputData, const Shape& inputShape, in localResponseNormFloat32Impl()
59 bool localResponseNormFloat16(const _Float16* inputData, const Shape& inputShape, int32_t radius, in localResponseNormFloat16()
74 bool localResponseNormFloat32(const float* inputData, const Shape& inputShape, int32_t radius, in localResponseNormFloat32()
|
| D | ArgMinMax.cpp | 30 static void argMinMaxImpl(const In* inputData, const Shape& inputShape, in argMinMaxImpl()
55 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, in argMinMaxGeneric()
|
| D | Softmax.cpp | 42 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
74 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
90 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
104 bool softmaxQuant8Impl(const uint8_t* inputData, const Shape& inputShape, const float beta, in softmaxQuant8Impl()
186 bool softmaxQuant8(const uint8_t* inputData, const Shape& inputShape, const float beta, in softmaxQuant8()
|
| D | ResizeImageOps.cpp | 47 bool resizeImageOpNhwc(OperationType opType, const T* inputData, const Shape& inputShape, in resizeImageOpNhwc()
76 const Shape& inputShape, _Float16* outputData, in resizeImageOpNhwc()
89 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
|
| D | TopK_V2.cpp | 29 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t k, T* valuesData, in evalGeneric()
67 bool eval(const void* inputData, const Shape& inputShape, const int32_t k, void* valuesData, in eval()
|
| D | SimpleMath.cpp | 51 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
66 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
|
| 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 | 47 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
77 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
98 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
|
| D | Tile.cpp | 65 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
86 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples, in eval()
|
| D | ExpandDims.cpp | 40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| D | Slice.cpp | 46 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t* beginData, in evalGeneric()
94 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| D | DepthwiseConv2D.cpp | 28 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape, in depthwiseConvFloat16()
65 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvFloat32()
98 bool depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape, in depthwiseConvQuant8()
146 bool depthwiseConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel()
|
| D | Conv2D.cpp | 94 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
164 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc()
189 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc()
236 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc()
263 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv()
282 bool convQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
375 bool convQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape, in convQuant8PerChannel()
|
| D | InstanceNormalization.cpp | 46 inline bool instanceNormNhwc(const T* inputData, const Shape& inputShape, T gamma, T beta, in instanceNormNhwc()
78 inline bool instanceNorm(const T* inputData, const Shape& inputShape, T gamma, T beta, T epsilon, in instanceNorm()
|
| D | TransposeConv2D.cpp | 115 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc()
171 bool transposeConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in transposeConvNhwc()
266 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc()
290 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv()
305 bool transposeConvQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in transposeConvQuant8PerChannelNhwc()
407 bool transposeConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape, in transposeConvQuant8PerChannel()
|
| D | UnidirectionalSequenceRNN.cpp | 41 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
59 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
|
| D | GroupedConv2D.cpp | 42 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in groupedConvFloat32()
99 bool groupedConvQuant8(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in groupedConvQuant8()
176 bool groupedConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape, in groupedConvQuant8PerChannel()
262 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape, in groupedConvFloat16()
|
| D | StridedSlice.cpp | 31 bool stridedSliceGeneric(const uint8_t* inputData, const Shape& inputShape, in stridedSliceGeneric()
|
| D | RoiPooling.cpp | 48 inline bool roiPoolingNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPoolingNhwc()
137 inline bool roiPooling(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPooling()
153 inline bool roiPooling<uint8_t, uint16_t>(const uint8_t* inputData, const Shape& inputShape, in roiPooling()
|