Searched refs:dot_as_convolution_util (Results 1 – 6 of 6) sorted by relevance
26 namespace dot_as_convolution_util {
25 namespace dot_as_convolution_util { namespace157 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()505 dot_as_convolution_util::ParseConvolutionDimsInfo(instruction); in InferConvolutionShardingFromOperands()858 dot_as_convolution_util::ParseDotGeneralFromDot(instruction); in InferShardingFromOperands()1047 const dot_as_convolution_util::DotConvolutionDimsInfo& dnums, in InferDotOperandSharding()1203 auto dot_dims = dot_as_convolution_util::ParseConvolutionDimsInfo(&user); in GetShardingFromUser()1305 auto dnums = dot_as_convolution_util::ParseDotGeneralFromDot(&user); in GetShardingFromUser()
520 ":dot_as_convolution_util",562 name = "dot_as_convolution_util",564 "dot_as_convolution_util.cc",567 "dot_as_convolution_util.h",
43 "//tensorflow/compiler/xla/service:dot_as_convolution_util",
895 const dot_as_convolution_util::DotConvolutionDimsInfo& dot_dnums, in CreateShardedConvConvolution()988 auto dims_info = dot_as_convolution_util::ParseConvolutionDimsInfo(hlo); in HandleConvolution()1032 dot_as_convolution_util::CreateShardedConvForDotGeneralConvolution( in HandleConvolution()