Searched refs:pyop_instance (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | generate_graph.cc | 176 …ValuePtr pyop_instance = CreatOpInstance(attrs, op_name, instance_name_base_ + op_name + std::to_s… in NewOpInst() local
177 if (pyop_instance == nullptr) { in NewOpInst()
180 auto value_node = NewValueNode(pyop_instance); in NewOpInst()
187 …ValuePtr pyop_instance = CreatOpInstance(attrs, op_name, instance_name_base_ + std::to_string(name… in NewOpInst() local
188 if (pyop_instance == nullptr) { in NewOpInst()
191 auto value_node = NewValueNode(pyop_instance); in NewOpInst()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 189 ValuePtr pyop_instance = CreatOpInstance(arg_replace_op.first, replace_op.first, instance_name); in ReplaceOpInput() local
190 if (pyop_instance == nullptr) { in ReplaceOpInput()
197 return {NewValueNode(pyop_instance)}; in ReplaceOpInput()
201 std::vector<AnfNodePtr> replace_input = {NewValueNode(pyop_instance), node->input(1)}; in ReplaceOpInput()
204 replace_input = {NewValueNode(pyop_instance), node->input(1), node->input(2)}; in ReplaceOpInput()
|
| D | step_parallel.cc | 104 ValuePtr pyop_instance = CreatOpInstance(arg_forward.first, op.first, instance_name); in CreateInput() local
105 MS_EXCEPTION_IF_NULL(pyop_instance); in CreateInput()
108 std::vector<AnfNodePtr> new_node_input = {NewValueNode(pyop_instance), node}; in CreateInput()
166 ValuePtr pyop_instance = CreatOpInstance(arg_forward.first, op_name, instance_name); in CreateMirrorInput() local
167 MS_EXCEPTION_IF_NULL(pyop_instance); in CreateMirrorInput()
173 new_node_input = {NewValueNode(pyop_instance), node, grad_accu}; in CreateMirrorInput()
176 new_node_input = {NewValueNode(pyop_instance), node}; in CreateMirrorInput()
|