Searched refs:ParseConvolutionDimsInfo (Results 1 – 4 of 4) sorted by relevance
52 DotConvolutionDimsInfo ParseConvolutionDimsInfo(const HloInstruction* conv);
52 /* static */ DotConvolutionDimsInfo ParseConvolutionDimsInfo( in ParseConvolutionDimsInfo() function
505 dot_as_convolution_util::ParseConvolutionDimsInfo(instruction); in InferConvolutionShardingFromOperands()1203 auto dot_dims = dot_as_convolution_util::ParseConvolutionDimsInfo(&user); in GetShardingFromUser()
988 auto dims_info = dot_as_convolution_util::ParseConvolutionDimsInfo(hlo); in HandleConvolution()