Searched defs:tuple_sharding (Results 1 – 4 of 4) sorted by relevance
174 HloSharding tuple_sharding = in TEST_F() local
358 for (auto& tuple_sharding : tuple_elements_) { in UniqueDevice() local
1056 xla::OpSharding tuple_sharding; in BuildArguments() local
1797 xla::HloSharding tuple_sharding = xla::HloSharding::Tuple( in TEST_F() local