Searched refs:sharding_elems (Results 1 – 1 of 1) sorted by relevance
371 std::vector<xla::HloSharding> sharding_elems; in BuildComputation() local385 sharding_elems.push_back(sub_sharding_elem); in BuildComputation()388 sharding_elems.push_back(sub_sharding); in BuildComputation()392 xla::HloSharding::Tuple(shape, sharding_elems); in BuildComputation()