Searched refs:is_shared_param (Results 1 – 2 of 2) sorted by relevance
119 void set_is_shared_param(bool is_shared_param) { is_shared_param_ = is_shared_param; } in set_is_shared_param() argument121 bool is_shared_param() { return is_shared_param_; } in is_shared_param() function
1153 bool is_shared_param = false; in InsertMirrorOps() local1163 is_shared_param = param_ptr->user_data<TensorLayout>()->is_shared_param(); in InsertMirrorOps()1197 if (pre_node->cast<CNodePtr>() && (InsertMirrorBeforeCast(node, index) || is_shared_param)) { in InsertMirrorOps()1519 … const AnfNodePtr &node, const std::string &op_name, bool is_shared_param) { in InsertAllGatherOp() argument1540 if (!is_shared_param && cast_node) { in InsertAllGatherOp()1601 bool is_shared_param = param_ptr->user_data<TensorLayout>()->is_shared_param(); in ApplyParallelOptOnParam() local1602 InsertAllGatherOp(root, opt_shard_group, param_pair, parameter, op_name, is_shared_param); in ApplyParallelOptOnParam()