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