Home
last modified time | relevance | path

Searched refs:filterScales (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/operations/
DGroupedConv2D.cpp202 const float* filterScales, const int32_t* biasData, in groupedConvQuant8PerChannel() argument
219 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel()
221 biasChannelShape.scale = filterScales[i] * inputShape.scale; in groupedConvQuant8PerChannel()
313 const Shape& filterShape, const float* filterScales, const int32_t* biasData,
320 const Shape& filterShape, const float* filterScales, const int32_t* biasData,
DConv2D.cpp328 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
356 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
358 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
421 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
449 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
451 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
484 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel() argument
495 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
DTransposeConv2D.cpp318 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument
346 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc()
348 biasChannelShape.scale = filterScales[i] * inputShape.scale; in transposeConvQuant8PerChannelNhwc()
421 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannel() argument
429 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
/frameworks/ml/nn/common/include/
DOperations.h71 const float* filterScales, const int32_t* biasData,
169 const float* filterScales, const int32_t* biasData,
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp2097 float filterScales[2] = {0.5f, 1.0f}; in convOpTest() local
2102 .scales = filterScales, in convOpTest()
2235 float filterScales[2] = {0.5f, 1.0f}; in depthwiseConvOpTest() local
2240 .scales = filterScales, in depthwiseConvOpTest()
3317 float filterScales[2] = {0.5f, 1.0f}; in groupedConvOpTest() local
3323 .scales = filterScales, in groupedConvOpTest()
3393 float filterScales[2] = {0.5f, 1.0f}; in transposeConvOpTest() local
3397 .scales = filterScales, in transposeConvOpTest()