Home
last modified time | relevance | path

Searched refs:filterChannelShape (Results 1 – 4 of 4) sorted by relevance

/frameworks/ml/nn/common/operations/
DGroupedConv2D.cpp218 Shape filterChannelShape = filterShape; in groupedConvQuant8PerChannel() local
219 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel()
224 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in groupedConvQuant8PerChannel()
DConv2D.cpp380 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local
381 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
385 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
473 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local
474 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc()
478 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
DTransposeConv2D.cpp345 Shape filterChannelShape = filterShape; in transposeConvQuant8PerChannelNhwc() local
346 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc()
351 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
DDepthwiseConv2D.cpp306 Shape filterChannelShape = filterShape; in depthwiseConvQuant8PerChannelNhwc() local
307 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannelNhwc()
311 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in depthwiseConvQuant8PerChannelNhwc()