Searched defs:device_sets (Results 1 – 2 of 2) sorted by relevance
81 std::vector<absl::Span<const int64_t>> device_sets; in DetermineHloInstructionIsReplicated() local193 std::vector<absl::Span<const int64_t>> device_sets; in DetermineHloInstructionIsReplicated() local536 absl::Span<const absl::Span<const int64_t>> device_sets) { in PartiallyReplicated()581 std::vector<absl::Span<const int64_t>> device_sets; in Merge() local
60 device_sets=(['/device:GPU:1', '/device:GPU:2', '/device:GPU:0'], argument