Searched refs:visiting_hlo_sharding_ (Results 1 – 2 of 2) sorted by relevance
548 absl::optional<HloSharding> visiting_hlo_sharding_; variable
1437 visiting_hlo_sharding_ = hlo->sharding(); in Preprocess()1439 manual_to_onedevice(hlo->shape(), *visiting_hlo_sharding_)); in Preprocess()1458 if (visiting_hlo_sharding_) { in Postprocess()1459 hlo->set_sharding(*visiting_hlo_sharding_); in Postprocess()1460 GetPartitionedHlo(hlo).hlo()->set_sharding(*visiting_hlo_sharding_); in Postprocess()1466 visiting_hlo_sharding_.reset(); in Postprocess()