| /packages/modules/NeuralNetworks/common/operations/ |
| D | GroupedConv2D.cpp | 46 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData, in groupedConvFloat32() argument
63 const float* filterBase = filterData; in groupedConvFloat32()
104 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData, in groupedConvQuant8() argument
135 const T* filterBase = filterData; in groupedConvQuant8()
182 const int8_t* filterData, const Shape& filterShape,
191 const uint8_t* filterData, const Shape& filterShape,
201 const int8_t* filterData, const Shape& filterShape, in groupedConvQuant8PerChannel() argument
239 const int8_t* filterBase = filterData; in groupedConvQuant8PerChannel()
286 const _Float16* filterData, const Shape& filterShape, in groupedConvFloat16() argument
299 convertFloat16ToFloat32(filterData, &filterData_float32); in groupedConvFloat16()
[all …]
|
| D | Conv2D.cpp | 194 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc() argument
215 inputData, convertShapeToDims(inputShape), filterData, convertShapeToDims(filterShape), in convNhwc()
223 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc() argument
264 tflite::optimized_ops::Conv(inputData, convertShapeToDims(inputShape), inputOffset, filterData, in convNhwc()
277 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc() argument
290 convertInt8ToUInt8(filterData, &unsignedFilter); in convNhwc()
306 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc() argument
320 convertFloat16ToFloat32(filterData, &filterData_float32); in convNhwc()
333 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv() argument
343 NN_RET_CHECK(convNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, in conv()
[all …]
|
| D | DepthwiseConv2D.cpp | 133 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc() argument
160 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
168 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc() argument
178 convertFloat16ToFloat32(filterData, &filterDataFloat32); in depthwiseConvNhwc()
193 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc() argument
236 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
244 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc() argument
258 convertInt8ToUInt8(filterData, &unsignedFilter); in depthwiseConvNhwc()
277 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc() argument
348 sum += (static_cast<int32_t>(filterData[filterIndex])) * in depthwiseConvQuant8PerChannelNhwc()
[all …]
|
| D | TransposeConv2D.cpp | 126 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc() argument
146 const float* filterBase = filterData; in transposeConvNhwc()
183 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc() argument
247 (static_cast<int32_t>(filterData[filterIndex]) + in transposeConvNhwc()
278 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc() argument
289 convertFloat16ToFloat32(filterData, &filterData_float32); in transposeConvNhwc()
301 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv() argument
309 NN_RET_CHECK(transposeConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in transposeConv()
318 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc() argument
389 static_cast<int32_t>(filterData[filterIndex]); in transposeConvQuant8PerChannelNhwc()
[all …]
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 53 const _Float16* filterData, const Shape& filterShape,
59 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
67 const uint8_t* filterData, const Shape& filterShape,
74 const int8_t* filterData, const Shape& filterShape,
149 const _Float16* filterData, const Shape& filterShape,
155 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
163 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData,
172 const int8_t* filterData, const Shape& filterShape,
|