Searched defs:require_matching_devices_to_group (Results 1 – 1 of 1) sorted by relevance
2015 bool require_matching_devices_to_group, in PartitionDotGroupOnBatch()2359 bool require_matching_devices_to_group, in PartitionDotGroupOnNonContracting()2588 bool require_matching_devices_to_group, in PartitionDotGroupOnContracting()2996 bool require_matching_devices_to_group, SpmdBuilder* b, in PrioritizeContractingDimensionsPartitioning()3351 bool require_matching_devices_to_group, in PartitionDot()