| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A 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,
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()
312 const uint8_t* inputData, const Shape& inputShape, const int8_t* filterData,
[all …]
|
| H A D | Conv2D.cpp | 194 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc() argument
223 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc() argument
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
353 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc() argument
400 const int8_t* filterBase = filterData; in convQuant8PerChannelNhwc()
446 const int8_t* filterData, const Shape& filterShape, in convQuant8PerChannelNhwc() argument
[all …]
|
| H A 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()
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()
384 NN_RET_CHECK(depthwiseConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in depthwiseConv()
[all …]
|
| H A 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()
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()
421 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel() argument
[all …]
|
| /aosp12/packages/modules/NeuralNetworks/common/include/ |
| H A 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,
|