| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | Activation.cpp | 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()
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()
159 bool tanhQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in tanhQuant8()
328 bool hardSwishQuant(const T* inputData, const Shape& inputShape, T* outputData, in hardSwishQuant()
[all …]
|
| H A D | Split.cpp | 29 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
55 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
62 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
69 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
76 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8Signed()
83 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
|
| H A D | Pooling.cpp | 81 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
141 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
164 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
174 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
187 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
197 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
209 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
219 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
229 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
241 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
[all …]
|
| H A 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()
|
| H A D | L2Normalization.cpp | 48 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis, in l2normFloat32Impl()
76 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl()
108 inline bool l2normQuant8SignedImpl(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8SignedImpl()
139 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
155 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16()
168 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8()
184 bool l2normQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Signed()
|
| H A D | TopK_V2.cpp | 41 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t k, T* valuesData, in evalGeneric()
96 const Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
114 const Shape inputShape = context->getInputShape(kInputTensor); in execute() local
|
| H A 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()
121 const Shape& inputShape = context->getInputShape(kInputTensor); in execute() local
|
| H A 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()
|
| H A D | Elu.cpp | 42 bool eluFloat(const T* inputData, const Shape& inputShape, const T alpha, T* outputData, in eluFloat()
73 Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| H A D | ArgMinMax.cpp | 29 static void argMinMaxImpl(const In* inputData, const Shape& inputShape, int32_t axis, bool isArgMin, in argMinMaxImpl()
51 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32 axis, bool isArgMin, in argMinMaxGeneric()
|
| H A D | Softmax.cpp | 53 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
85 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
101 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
116 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8Impl()
203 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8()
|
| H A D | FullyConnected.cpp | 56 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
86 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
107 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
148 bool fullyConnectedQuant8(const int8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
|
| H A D | RNN.cpp | 68 const Shape& inputShape = input->shape(); in Prepare() local
116 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep()
136 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep()
|
| H A D | ChannelShuffle.cpp | 38 inline bool eval(const T* inputData, const Shape& inputShape, int32_t numGroups, int32_t axis, in eval()
69 const Shape& inputShape = context->getInputShape(kInputTensor); in validate() local
|
| H A D | Tile.cpp | 69 void tileImpl(const T* inputData, const Shape& inputShape, const int32_t* multiples, T* outputData, in tileImpl()
90 bool eval(const uint8_t* inputData, const Shape& inputShape, const int32_t* multiples, in eval()
|
| H A D | Conv2D.cpp | 107 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
194 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc()
223 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc()
277 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc()
306 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc()
333 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv()
352 bool convQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
445 bool convQuant8PerChannelNhwc(const int8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
509 bool convQuant8PerChannel(const T* inputData, const Shape& inputShape, const int8_t* filterData, in convQuant8PerChannel()
|
| H A D | DepthwiseConv2D.cpp | 96 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
133 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc()
167 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc()
193 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc()
244 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc()
277 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc()
373 bool depthwiseConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in depthwiseConv()
394 bool depthwiseConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel()
|
| H A D | ExpandDims.cpp | 40 bool eval(const uint8_t* inputData, const Shape& inputShape, int32_t axis, uint8_t* outputData, in eval()
|
| H A D | Slice.cpp | 53 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t* beginData, in evalGeneric()
111 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| H A D | ResizeImageOps.cpp | 68 bool resizeNearestNeighbor(const T* inputData, const Shape& inputShape, bool alignCorners, in resizeNearestNeighbor()
111 bool resizeImageOpNhwc(OperationType opType, const T* inputData, const Shape& inputShape, in resizeImageOpNhwc()
152 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
|
| H A D | InstanceNormalization.cpp | 49 inline bool instanceNormNhwc(const T* inputData, const Shape& inputShape, T gamma, T beta, in instanceNormNhwc()
92 inline bool instanceNorm(const T* inputData, const Shape& inputShape, T gamma, T beta, T epsilon, in instanceNorm()
|
| H A D | TransposeConv2D.cpp | 126 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc()
183 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc()
277 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc()
301 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv()
317 bool transposeConvQuant8PerChannelNhwc(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannelNhwc()
420 bool transposeConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannel()
|
| H A D | GroupedConv2D.cpp | 46 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in groupedConvFloat32()
104 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData, in groupedConvQuant8()
200 bool groupedConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in groupedConvQuant8PerChannel()
285 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape, in groupedConvFloat16()
|
| H A D | StridedSlice.cpp | 53 bool compute(const T* inputData, const Shape& inputShape, const int32_t* beginData, in compute()
143 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| H A D | UnidirectionalSequenceRNN.cpp | 49 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
67 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
|