Searched defs:device_set_root (Results 1 – 1 of 1) sorted by relevance
518 absl::Span<const int64_t> device_set_root) in HloReplication()543 std::vector<int64_t> device_set_root; in PartiallyReplicated() local