Searched refs:set_opt_weight_shard_step (Results 1 – 3 of 3) sorted by relevance
111 void set_opt_weight_shard_step(int32_t step) { opt_weight_shard_step_ = step; } in set_opt_weight_shard_step() function
188 parallel_layouts->set_opt_weight_shard_step(tensor_layout->opt_weight_shard_step()); in Save()
612 tensor_layout->set_opt_weight_shard_step(LongToInt(opt_weight_shard_step)); in CreateGroupForOptShard()