Searched refs:entry_params_shardings (Results 1 – 1 of 1) sorted by relevance
3658 std::vector<HloSharding> entry_params_shardings; in Run() local3662 entry_params_shardings.push_back(param->sharding()); in Run()3664 module->set_spmd_parameters_shardings(entry_params_shardings); in Run()