| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Activation.cpp | 54 bool reluFloat(const T* inputData, const Shape& inputShape, T* outputData, in reluFloat()
72 bool relu1Float(const T* inputData, const Shape& inputShape, T* outputData, in relu1Float()
82 bool relu6Float(const T* inputData, const Shape& inputShape, T* outputData, in relu6Float()
91 bool tanhFloat16(const _Float16* inputData, const Shape& inputShape, _Float16* outputData, in tanhFloat16()
101 bool tanhFloat32(const float* inputData, const Shape& inputShape, float* outputData, in tanhFloat32()
112 bool logisticFloat(const T* inputData, const Shape& inputShape, T* outputData, in logisticFloat()
127 inline bool reluXQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluXQuant8()
143 bool reluQuant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in reluQuant8()
149 bool relu1Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu1Quant8()
155 bool relu6Quant8(const uint8_t* inputData, const Shape& inputShape, uint8_t* outputData, in relu6Quant8()
[all …]
|
| D | Pooling.cpp | 83 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
143 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
153 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
166 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
176 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
189 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
199 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
211 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
221 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
231 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
[all …]
|
| D | Split.cpp | 31 bool splitGeneric(const Scalar* inputData, const Shape& inputShape, int32_t axis, in splitGeneric()
57 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in splitFloat16()
64 bool splitFloat32(const float* inputData, const Shape& inputShape, int32_t axis, in splitFloat32()
71 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8()
78 bool splitQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in splitQuant8Signed()
85 bool splitInt32(const int32_t* inputData, const Shape& inputShape, int32_t axis, in splitInt32()
|
| D | Reshape.cpp | 40 bool copyData(const void* inputData, const Shape& inputShape, void* outputData, in copyData()
49 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in depthToSpaceGeneric()
70 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize, in spaceToDepthGeneric()
91 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T padValue, in padGeneric()
204 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in batchToSpaceGeneric()
229 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize, in spaceToBatchGeneric()
|
| D | TopK_V2.cpp | 34 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t k, T* valuesData, in evalGeneric()
70 const Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
88 const Shape inputShape = context->getInputShape(kInputTensor); in execute() local
|
| D | L2Normalization.cpp | 46 inline bool l2normFloat32Impl(const float* inputData, const Shape& inputShape, int32_t axis, in l2normFloat32Impl()
74 inline bool l2normQuant8Impl(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Impl()
106 inline bool l2normQuant8SignedImpl(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8SignedImpl()
137 bool l2normFloat32(const float* inputData, const Shape& inputShape, int32_t axis, float* outputData, in l2normFloat32()
153 bool l2normFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis, in l2normFloat16()
166 bool l2normQuant8(const uint8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8()
182 bool l2normQuant8Signed(const int8_t* inputData, const Shape& inputShape, int32_t axis, in l2normQuant8Signed()
|
| D | Dequantize.cpp | 31 bool compute(const InputType* inputData, const Shape& inputShape, OutputType* outputData) { in compute()
44 bool computePerChannel(const int8_t* inputData, const Shape& inputShape, OutputType* outputData) { in computePerChannel()
89 const Shape& inputShape = context->getInputShape(kInputTensor); in execute() local
|
| D | Elu.cpp | 36 bool eluFloat(const T* inputData, const Shape& inputShape, const T alpha, T* outputData, in eluFloat()
50 Shape inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| D | FullyConnected.cpp | 52 bool fullyConnectedFloat32(const float* inputData, const Shape& inputShape, in fullyConnectedFloat32()
82 bool fullyConnectedFloat16(const _Float16* inputData, const Shape& inputShape, in fullyConnectedFloat16()
103 bool fullyConnectedQuant8(const uint8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
144 bool fullyConnectedQuant8(const int8_t* inputData, const Shape& inputShape, in fullyConnectedQuant8()
|
| D | Softmax.cpp | 50 inline bool softmaxSlowFloat32(const float* inputData, const Shape& inputShape, const float beta, in softmaxSlowFloat32()
82 bool softmaxFloat32(const float* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxFloat32()
98 bool softmaxFloat16(const _Float16* inputData, const Shape& inputShape, const float beta, in softmaxFloat16()
113 bool softmaxQuant8Impl(const T* inputData, const Shape& inputShape, const float /*beta*/, in softmaxQuant8Impl()
200 bool softmaxQuant8(const T* inputData, const Shape& inputShape, const float beta, int32_t axis, in softmaxQuant8()
|
| D | SimpleMath.cpp | 39 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis, in meanFloat16()
54 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape, in meanGeneric()
|
| D | ResizeImageOps.cpp | 61 bool resizeNearestNeighbor(const T* inputData, const Shape& inputShape, bool alignCorners, in resizeNearestNeighbor()
104 bool resizeImageOpNhwc(OperationType opType, const T* inputData, const Shape& inputShape, in resizeImageOpNhwc()
132 const Shape& inputShape, bool alignCorners, bool halfPixelCenters, in resizeImageOpNhwc()
145 bool resizeImageOp(OperationType opType, const T* inputData, const Shape& inputShape, bool useNchw, in resizeImageOp()
|
| D | ArgMinMax.cpp | 31 static void argMinMaxImpl(const In* inputData, const Shape& inputShape, int32_t axis, bool isArgMin, in argMinMaxImpl()
53 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32 axis, bool isArgMin, in argMinMaxGeneric()
|
| D | LocalResponseNormalization.cpp | 45 inline bool localResponseNormFloat32Impl(const float* inputData, const Shape& inputShape, in localResponseNormFloat32Impl()
80 bool localResponseNorm<float>(const float* inputData, const Shape& inputShape, int32_t radius, in localResponseNorm()
102 bool localResponseNorm<_Float16>(const _Float16* inputData, const Shape& inputShape, int32_t radius, in localResponseNorm()
|
| D | RoiPooling.cpp | 42 inline bool roiPoolingNhwc(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPoolingNhwc()
131 inline bool roiPooling(const T_Input* inputData, const Shape& inputShape, const T_Roi* roiData, in roiPooling()
147 inline bool roiPooling<uint8_t, uint16_t>(const uint8_t* inputData, const Shape& inputShape, in roiPooling()
162 inline bool roiPooling<int8_t, uint16_t>(const int8_t* inputData, const Shape& inputShape, in roiPooling()
|
| 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()
|
| D | Conv2D.cpp | 105 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
191 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc()
220 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc()
274 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc()
303 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc()
330 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv()
349 bool convQuant8PerChannelNhwc(const uint8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
442 bool convQuant8PerChannelNhwc(const int8_t* inputData, const Shape& inputShape, in convQuant8PerChannelNhwc()
506 bool convQuant8PerChannel(const T* inputData, const Shape& inputShape, const int8_t* filterData, in convQuant8PerChannel()
|
| D | DepthwiseConv2D.cpp | 92 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
129 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc()
163 bool depthwiseConvNhwc(const _Float16* inputData, const Shape& inputShape, in depthwiseConvNhwc()
189 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc()
240 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc()
273 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc()
369 bool depthwiseConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in depthwiseConv()
390 bool depthwiseConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in depthwiseConvQuant8PerChannel()
|
| D | Slice.cpp | 45 bool evalGeneric(const T* inputData, const Shape& inputShape, const int32_t* beginData, in evalGeneric()
78 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| D | InstanceNormalization.cpp | 39 inline bool instanceNormNhwc(const T* inputData, const Shape& inputShape, T gamma, T beta, in instanceNormNhwc()
82 inline bool instanceNorm(const T* inputData, const Shape& inputShape, T gamma, T beta, T epsilon, in instanceNorm()
|
| D | StridedSlice.cpp | 47 bool compute(const T* inputData, const Shape& inputShape, const int32_t* beginData, in compute()
99 const Shape& inputShape = context->getInputShape(kInputTensor); in prepare() local
|
| 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()
|
| D | TransposeConv2D.cpp | 122 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc()
179 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc()
273 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc()
297 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv()
313 bool transposeConvQuant8PerChannelNhwc(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannelNhwc()
416 bool transposeConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannel()
|
| D | UnidirectionalSequenceRNN.cpp | 37 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims()
55 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped() local
|
| /packages/modules/NeuralNetworks/common/types/operations/src/ |
| D | ChannelShuffle.cpp | 33 const Shape& inputShape = context->getInputShape(kInputTensor); in validate() local
|