Searched refs:opt_shard_mirror_group (Results 1 – 2 of 2) sorted by relevance
109 std::string opt_shard_mirror_group() { return opt_shard_mirror_group_; } in opt_shard_mirror_group() function
920 …if (param_ptr && !param_ptr->opt_shard_group().empty() && param_ptr->opt_shard_mirror_group().empt… in FindParameterByValueNode()930 …if (param_ptr && !param_ptr->opt_shard_group().empty() && param_ptr->opt_shard_mirror_group().empt… in FindParameterByParameter()1160 std::string opt_shard_mirror_group; in InsertMirrorOps() local1162 opt_shard_mirror_group = param_ptr->user_data<TensorLayout>()->opt_shard_mirror_group(); in InsertMirrorOps()1165 if (!opt_shard_mirror_group.empty()) { in InsertMirrorOps()1167 …backward_op = CreateMirrorOps(opt_shard_mirror_group, static_cast<size_t>(opt_shard_mirror_group[0… in InsertMirrorOps()