Searched refs:dims_info (Results 1 – 1 of 1) sorted by relevance
988 auto dims_info = dot_as_convolution_util::ParseConvolutionDimsInfo(hlo); in HandleConvolution() local990 for (const auto& dims : dims_info.batch_dims) { in HandleConvolution()997 for (const auto& dims : dims_info.contracting_dims) { in HandleConvolution()1004 for (const auto& dims : dims_info.lhs_non_contracting_dims) { in HandleConvolution()1011 for (const auto& dims : dims_info.rhs_non_contracting_dims) { in HandleConvolution()1018 for (const auto& dims : dims_info.conv_spatial_dims) { in HandleConvolution()1029 if (dims_info.conv_spatial_dims.empty()) { in HandleConvolution()1033 *hlo, dims_info, lhs_hlo, rhs_hlo)); in HandleConvolution()1037 CreateShardedConvConvolution(*hlo, dims_info, lhs_hlo, in HandleConvolution()