Searched refs:require_matching_devices_to_group (Results 1 – 1 of 1) sorted by relevance
1658 bool require_matching_devices_to_group, in PartitionDotGroupOnBatch() argument1698 if (require_matching_devices_to_group && lhs_rhs_dims_matching) { in PartitionDotGroupOnBatch()1813 if (require_matching_devices_to_group && in PartitionDotGroupOnBatch()1992 bool require_matching_devices_to_group, in PartitionDotGroupOnNonContracting() argument2014 if (require_matching_devices_to_group && in PartitionDotGroupOnNonContracting()2087 bool require_matching_devices_to_group, in PartitionDotGroupOnContracting() argument2136 if (require_matching_devices_to_group && rhs.sharding() != rhs_sharding) { in PartitionDotGroupOnContracting()2143 if (require_matching_devices_to_group && lhs.sharding() != lhs_sharding) { in PartitionDotGroupOnContracting()2439 bool require_matching_devices_to_group, in PartitionDot() argument2575 require_matching_devices_to_group, options, b, in PartitionDot()[all …]