Searched refs:filterChannelShape (Results 1 – 4 of 4) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | DepthwiseConv2D.cpp | 180 Shape filterChannelShape = filterShape; in depthwiseConvQuant8PerChannel() local
181 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannel()
185 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in depthwiseConvQuant8PerChannel()
|
| D | GroupedConv2D.cpp | 195 Shape filterChannelShape = filterShape; in groupedConvQuant8PerChannel() local
196 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel()
201 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in groupedConvQuant8PerChannel()
|
| D | TransposeConv2D.cpp | 334 Shape filterChannelShape = filterShape; in transposeConvQuant8PerChannelNhwc() local
335 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc()
340 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
|
| D | Conv2D.cpp | 311 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local
312 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
316 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
|