Searched refs:DotConvolutionDimsInfo (Results 1 – 4 of 4) sorted by relevance
30 struct DotConvolutionDimsInfo { struct52 DotConvolutionDimsInfo ParseConvolutionDimsInfo(const HloInstruction* conv); argument62 const HloInstruction& conv, const DotConvolutionDimsInfo& dot_dnums,73 DotConvolutionDimsInfo ParseDotGeneralFromDot(const HloInstruction* dot);
52 /* static */ DotConvolutionDimsInfo ParseConvolutionDimsInfo( in ParseConvolutionDimsInfo()56 DotConvolutionDimsInfo dims; in ParseConvolutionDimsInfo()110 const HloInstruction& conv, const DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvForDotGeneralConvolution()155 DotConvolutionDimsInfo ParseDotGeneralFromDot(const HloInstruction* dot) { in ParseDotGeneralFromDot()157 dot_as_convolution_util::DotConvolutionDimsInfo dnums; in ParseDotGeneralFromDot()
338 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotShardingFromOperands()483 const dot_as_convolution_util::DotConvolutionDimsInfo::DimNums> in InferConvolutionShardingFromOperands()1047 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotOperandSharding()
895 const dot_as_convolution_util::DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvConvolution()