Searched refs:filterScales (Results 1 – 6 of 6) sorted by relevance
/packages/modules/NeuralNetworks/common/operations/ |
D | GroupedConv2D.cpp | 202 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,
|
D | Conv2D.cpp | 354 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument 382 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc() 384 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc() 447 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument 475 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc() 477 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc() 510 const Shape& filterShape, const float* filterScales, in convQuant8PerChannel() argument 521 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
|
D | TransposeConv2D.cpp | 319 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument 347 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc() 349 biasChannelShape.scale = filterScales[i] * inputShape.scale; in transposeConvQuant8PerChannelNhwc() 422 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannel() argument 430 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
|
D | DepthwiseConv2D.cpp | 278 const Shape& filterShape, const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannelNhwc() argument 309 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannelNhwc() 311 biasChannelShape.scale = filterScales[i] * inputShape.scale; in depthwiseConvQuant8PerChannelNhwc() 396 const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannel() argument 408 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in depthwiseConvQuant8PerChannel()
|
/packages/modules/NeuralNetworks/common/include/ |
D | Operations.h | 75 const float* filterScales, const int32_t* biasData, 173 const float* filterScales, const int32_t* biasData,
|
/packages/modules/NeuralNetworks/runtime/test/ |
D | TestValidateOperations.cpp | 2097 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()
|