Searched refs:rhs_conv_spatial_partitions (Results 1 – 2 of 2) sorted by relevance
563 const int64_t rhs_conv_spatial_partitions = get_partitions_for_dims( in InferConvolutionShardingFromOperands() local566 (lhs_conv_spatial_partitions == 1 && rhs_conv_spatial_partitions == 1 && in InferConvolutionShardingFromOperands()
3405 const int64_t rhs_conv_spatial_partitions = get_partitions_for_dims( in PartitionDot() local3416 (lhs_conv_spatial_partitions > 1 || rhs_conv_spatial_partitions > 1 || in PartitionDot()