Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DGroupedConv2D.cpp206 const float* filterScales, const int32_t* biasData, in groupedConvQuant8PerChannel() argument
224 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel()
226 biasChannelShape.scale = filterScales[i] * inputShape.scale; in groupedConvQuant8PerChannel()
318 const Shape& filterShape, const float* filterScales, const int32_t* biasData,
325 const Shape& filterShape, const float* filterScales, const int32_t* biasData,
DConv2D.cpp351 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
379 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
381 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
444 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
472 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
474 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
507 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel() argument
518 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
DTransposeConv2D.cpp315 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument
343 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc()
345 biasChannelShape.scale = filterScales[i] * inputShape.scale; in transposeConvQuant8PerChannelNhwc()
418 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannel() argument
426 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
DDepthwiseConv2D.cpp274 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc() argument
305 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannelNhwc()
307 biasChannelShape.scale = filterScales[i] * inputShape.scale; in depthwiseConvQuant8PerChannelNhwc()
392 const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannel() argument
404 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in depthwiseConvQuant8PerChannel()
/packages/modules/NeuralNetworks/common/include/
DOperations.h58 const float* filterScales, const int32_t* biasData,
156 const float* filterScales, const int32_t* biasData,
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp2136 float filterScales[2] = {0.5f, 1.0f}; in convOpTest() local
2141 .scales = filterScales, in convOpTest()
2274 float filterScales[2] = {0.5f, 1.0f}; in depthwiseConvOpTest() local
2279 .scales = filterScales, in depthwiseConvOpTest()
3356 float filterScales[2] = {0.5f, 1.0f}; in groupedConvOpTest() local
3362 .scales = filterScales, in groupedConvOpTest()
3432 float filterScales[2] = {0.5f, 1.0f}; in transposeConvOpTest() local
3436 .scales = filterScales, in transposeConvOpTest()