Searched refs:optimizer_weight_shard_aggregated_save (Results 1 – 5 of 5) sorted by relevance
104 void set_optimizer_weight_shard_aggregated_save(bool optimizer_weight_shard_aggregated_save);105 …bool optimizer_weight_shard_aggregated_save() const { return optimizer_weight_shard_aggregated_sav… in optimizer_weight_shard_aggregated_save() function
149 …lContext::set_optimizer_weight_shard_aggregated_save(bool optimizer_weight_shard_aggregated_save) { in set_optimizer_weight_shard_aggregated_save() argument150 optimizer_weight_shard_aggregated_save_ = optimizer_weight_shard_aggregated_save; in set_optimizer_weight_shard_aggregated_save()
572 def set_optimizer_weight_shard_aggregated_save(self, optimizer_weight_shard_aggregated_save): argument581 if not isinstance(optimizer_weight_shard_aggregated_save, bool):583 …._context_handle.set_optimizer_weight_shard_aggregated_save(optimizer_weight_shard_aggregated_save)686 optimizer_weight_shard_aggregated_save=bool,
197 …_optimizer_weight_shard_aggregated_save", &ParallelContext::optimizer_weight_shard_aggregated_save, in PYBIND11_MODULE()
607 auto integrated_save = ParallelContext::GetInstance()->optimizer_weight_shard_aggregated_save(); in CreateGroupForOptShard()