Searched refs:filterScales (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 148 const float* filterScales, const int32_t* biasData, in depthwiseConvQuant8PerChannel() argument
181 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannel()
183 biasChannelShape.scale = filterScales[i] * inputShape.scale; in depthwiseConvQuant8PerChannel()
|
| D | GroupedConv2D.cpp | 178 const float* filterScales, const int32_t* biasData, in groupedConvQuant8PerChannel() argument
196 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel()
198 biasChannelShape.scale = filterScales[i] * inputShape.scale; in groupedConvQuant8PerChannel()
|
| D | TransposeConv2D.cpp | 307 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannelNhwc() argument
335 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc()
337 biasChannelShape.scale = filterScales[i] * inputShape.scale; in transposeConvQuant8PerChannelNhwc()
409 const float* filterScales, const int32_t* biasData, in transposeConvQuant8PerChannel() argument
417 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in transposeConvQuant8PerChannel()
|
| D | Conv2D.cpp | 284 const float* filterScales, const int32_t* biasData, in convQuant8PerChannelNhwc() argument
312 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
314 biasChannelShape.scale = filterScales[i] * inputShape.scale; in convQuant8PerChannelNhwc()
377 const float* filterScales, const int32_t* biasData, in convQuant8PerChannel() argument
388 input.getNhwcBuffer(), input.getNhwcShape(), filterData, filterShape, filterScales, in convQuant8PerChannel()
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 72 const float* filterScales, const int32_t* biasData,
164 const float* filterScales, const int32_t* biasData,
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1229 float filterScales[2] = {0.5f, 1.0f}; in convOpTest() local
1234 .scales = filterScales, in convOpTest()
1352 float filterScales[2] = {0.5f, 1.0f}; in depthwiseConvOpTest() local
1357 .scales = filterScales, in depthwiseConvOpTest()
2376 float filterScales[2] = {0.5f, 1.0f}; in groupedConvOpTest() local
2382 .scales = filterScales, in groupedConvOpTest()
2441 float filterScales[2] = {0.5f, 1.0f}; in transposeConvOpTest() local
2445 .scales = filterScales, in transposeConvOpTest()
|