| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 49 const _Float16* filterData, const Shape& filterShape,
55 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
63 const uint8_t* filterData, const Shape& filterShape,
70 const int8_t* filterData, const Shape& filterShape,
145 const _Float16* filterData, const Shape& filterShape,
151 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
159 bool groupedConvQuant8(const T* inputData, const Shape& inputShape, const T* filterData,
168 const int8_t* filterData, const Shape& filterShape,
|
| /frameworks/ml/nn/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 | 193 bool convNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in convNhwc() argument
214 inputData, convertShapeToDims(inputShape), filterData, convertShapeToDims(filterShape), in convNhwc()
222 bool convNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in convNhwc() argument
263 tflite::optimized_ops::Conv(inputData, convertShapeToDims(inputShape), inputOffset, filterData, in convNhwc()
276 bool convNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in convNhwc() argument
289 convertInt8ToUInt8(filterData, &unsignedFilter); in convNhwc()
305 bool convNhwc(const _Float16* inputData, const Shape& inputShape, const _Float16* filterData, in convNhwc() argument
319 convertFloat16ToFloat32(filterData, &filterData_float32); in convNhwc()
332 bool conv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in conv() argument
342 NN_RET_CHECK(convNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, in conv()
[all …]
|
| D | DepthwiseConv2D.cpp | 131 bool depthwiseConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in depthwiseConvNhwc() argument
158 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
166 const _Float16* filterData, const Shape& filterShape, in depthwiseConvNhwc() argument
176 convertFloat16ToFloat32(filterData, &filterDataFloat32); in depthwiseConvNhwc()
191 bool depthwiseConvNhwc(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData, in depthwiseConvNhwc() argument
234 convertShapeToTflshape(filterShape), filterData, in depthwiseConvNhwc()
242 bool depthwiseConvNhwc(const int8_t* inputData, Shape inputShape, const int8_t* filterData, in depthwiseConvNhwc() argument
256 convertInt8ToUInt8(filterData, &unsignedFilter); in depthwiseConvNhwc()
275 const T* inputData, const Shape& inputShape, const int8_t* filterData, in depthwiseConvQuant8PerChannelNhwc() argument
346 sum += (static_cast<int32_t>(filterData[filterIndex])) * in depthwiseConvQuant8PerChannelNhwc()
[all …]
|
| D | TransposeConv2D.cpp | 125 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc() argument
145 const float* filterBase = filterData; in transposeConvNhwc()
182 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc() argument
246 (static_cast<int32_t>(filterData[filterIndex]) + in transposeConvNhwc()
277 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc() argument
288 convertFloat16ToFloat32(filterData, &filterData_float32); in transposeConvNhwc()
300 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv() argument
308 NN_RET_CHECK(transposeConvNhwc(input.getNhwcBuffer(), input.getNhwcShape(), filterData, in transposeConv()
317 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc() argument
388 static_cast<int32_t>(filterData[filterIndex]); in transposeConvQuant8PerChannelNhwc()
[all …]
|