Searched refs:opt_weight_shard_step (Results 1 – 5 of 5) sorted by relevance
57 required int32 opt_weight_shard_step = 6; field
113 int32_t opt_weight_shard_step() { return opt_weight_shard_step_; } in opt_weight_shard_step() function
188 parallel_layouts->set_opt_weight_shard_step(tensor_layout->opt_weight_shard_step()); in Save()
610 int64_t opt_weight_shard_step = in CreateGroupForOptShard() local612 tensor_layout->set_opt_weight_shard_step(LongToInt(opt_weight_shard_step)); in CreateGroupForOptShard()613 …MS_LOG(INFO) << "Parallel optimizer: save opt_weight_shard_step " << opt_weight_shard_step << " in… in CreateGroupForOptShard()
1499 shard_stride = int(layout.opt_weight_shard_step)