Home
last modified time | relevance | path

Searched refs:filterScales (Results 1 – 6 of 6) 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.cpp353 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
381 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
383 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
446 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
474 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
476 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
509 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel() argument
520 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()
DDepthwiseConv2D.cpp276 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc() argument
307 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannelNhwc()
309 biasChannelShape.scale = filterScales[i] * inputShape.scale; in depthwiseConvQuant8PerChannelNhwc()
394 const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannel() argument
406 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in depthwiseConvQuant8PerChannel()
/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.cpp2098 float filterScales[2] = {0.5f, 1.0f}; in convOpTest() local
2103 .scales = filterScales, in convOpTest()
2236 float filterScales[2] = {0.5f, 1.0f}; in depthwiseConvOpTest() local
2241 .scales = filterScales, in depthwiseConvOpTest()
3318 float filterScales[2] = {0.5f, 1.0f}; in groupedConvOpTest() local
3324 .scales = filterScales, in groupedConvOpTest()
3394 float filterScales[2] = {0.5f, 1.0f}; in transposeConvOpTest() local
3398 .scales = filterScales, in transposeConvOpTest()