Searched refs:filterChannelShape (Results 1 – 4 of 4) sorted by relevance
/packages/modules/NeuralNetworks/common/cpu_operations/ |
D | Conv2D.cpp | 378 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local 379 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc() 383 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc() 471 Shape filterChannelShape = filterShape; in convQuant8PerChannelNhwc() local 472 filterChannelShape.scale = filterScales[i]; in convQuant8PerChannelNhwc() 476 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in convQuant8PerChannelNhwc()
|
D | GroupedConv2D.cpp | 223 Shape filterChannelShape = filterShape; in groupedConvQuant8PerChannel() local 224 filterChannelShape.scale = filterScales[i]; in groupedConvQuant8PerChannel() 229 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in groupedConvQuant8PerChannel()
|
D | TransposeConv2D.cpp | 342 Shape filterChannelShape = filterShape; in transposeConvQuant8PerChannelNhwc() local 343 filterChannelShape.scale = filterScales[i]; in transposeConvQuant8PerChannelNhwc() 348 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in transposeConvQuant8PerChannelNhwc()
|
D | DepthwiseConv2D.cpp | 304 Shape filterChannelShape = filterShape; in depthwiseConvQuant8PerChannelNhwc() local 305 filterChannelShape.scale = filterScales[i]; in depthwiseConvQuant8PerChannelNhwc() 309 inputShape, filterChannelShape, biasChannelShape, outputShape, &realMultiplier[i])); in depthwiseConvQuant8PerChannelNhwc()
|