Searched refs:filterChannelShape (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | GroupedConv2D.cpp | 218 Shape filterChannelShape = filterShape; in groupedConvQuant8PerChannel() local
219 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel()
224 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in groupedConvQuant8PerChannel()
|
| D | Conv2D.cpp | 355 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local
356 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
360 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
448 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local
449 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
453 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
|
| D | TransposeConv2D.cpp | 345 Shape filterChannelShape = filterShape; in transposeConvQuant8PerChannelNhwc() local
346 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc()
351 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
|