Searched refs:rhs_conv_spatial_partitions (Results 1 – 2 of 2) sorted by relevance
508 const int64 rhs_conv_spatial_partitions = get_partitions_for_dims( in InferConvolutionShardingFromOperands() local511 (lhs_conv_spatial_partitions == 1 && rhs_conv_spatial_partitions == 1 && in InferConvolutionShardingFromOperands()
2492 const int64 rhs_conv_spatial_partitions = get_partitions_for_dims( in PartitionDot() local2502 (lhs_conv_spatial_partitions > 1 || rhs_conv_spatial_partitions > 1 || in PartitionDot()