Searched defs:assign_node (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/ |
| D | depend_elimination.cc | 53 const auto &assign_node = inputs[kDependAttachNodeIndex]->cast<CNodePtr>(); in Run() local
69 auto assign_node = VectorRef({prim::kPrimAssign, input1_, reducesum_node}); in DefinePattern() local
86 auto assign_node = node->cast<CNodePtr>()->input(kIndex2); in Process() local
|
| D | add_stitch_atomic_clean_gpu.cc | 76 …auto assign_node = CreateCNode({NewValueNode(prim::kPrimAssign), new_parameter, out_node}, sub_gra… in CreateAssignNode() local
|
| D | tensor_inplace.cc | 106 …auto assign_node = std::find_if(users.begin(), users.end(), [](const std::pair<AnfNodePtr, int> &p… in ReduceChecker() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | output_variable_pass.cc | 94 auto assign_node = std::make_shared<ops::Assign>(); in Run() local
|
| D | input_and_output_variable_pass.cc | 127 auto assign_node = std::make_shared<ops::Assign>(); in CreateAssign() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/operator/composite/ |
| D | starred_operation.cc | 80 auto assign_node = ret_graph->add_parameter(); in GenerateFuncGraph() local
|
| D | composite.cc | 1926 auto assign_node = GetAssignNode(graph, assign_parameter, step_value); in BuildFuncGraph() local
1987 …ePtr ListSliceSetItem::GetAssignNode(const FuncGraphPtr &func_graph, const AnfNodePtr &assign_node, in GetAssignNode()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/ |
| D | function_block.cc | 762 auto assign_node = func_graph_->NewCNodeInOrder({assign_op, target, source}); in SetStateAssign() local
|