Searched refs:visiting_hlo_operand_shardings_ (Results 1 – 2 of 2) sorted by relevance
547 std::vector<HloSharding> visiting_hlo_operand_shardings_; variable
1441 visiting_hlo_operand_shardings_.reserve(hlo->operand_count()); in Preprocess()1443 visiting_hlo_operand_shardings_.push_back(operand->sharding()); in Preprocess()1463 operand->set_sharding(visiting_hlo_operand_shardings_[i]); in Postprocess()1467 visiting_hlo_operand_shardings_.clear(); in Postprocess()