Home
last modified time | relevance | path

Searched refs:new_value_node (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/core/load_mindir/
Danf_model_parser.cc599 auto new_value_node = NewValueNode(MakeValue(tensor_info)); in ObtainValueNodeInTensorForm() local
600 MS_EXCEPTION_IF_NULL(new_value_node); in ObtainValueNodeInTensorForm()
603 new_value_node->set_abstract(tensor_abstract); in ObtainValueNodeInTensorForm()
604 anfnode_build_map_[value_node_name] = new_value_node; in ObtainValueNodeInTensorForm()
629 auto new_value_node = NewValueNode(MakeValue(tensor_vec)); in ObtainValueNodeInTupleTensorForm() local
630 anfnode_build_map_[value_node_name] = new_value_node; in ObtainValueNodeInTupleTensorForm()
641 auto new_value_node = NewValueNode(TypeIdToType(kDefaultValueSwitchMap[attr_tensor_type])); in ObtainValueNodeInTypeForm() local
643 new_value_node->set_abstract(abs_type); in ObtainValueNodeInTypeForm()
644 anfnode_build_map_[value_node_name] = new_value_node; in ObtainValueNodeInTypeForm()
649 auto new_value_node = NewValueNode(kNone); in ObtainValueNodeInNoneForm() local
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dkernel_build_ascend.cc166 auto new_value_node = NewValueNode(clear_zero_prim); in AddTbeClearZeroNode() local
167 MS_EXCEPTION_IF_NULL(new_value_node); in AddTbeClearZeroNode()
168 std::vector<AnfNodePtr> inputs = {new_value_node}; in AddTbeClearZeroNode()
193 auto new_value_node = NewValueNode(clear_zero_prim); in AddFusionTbeClearZeroNode() local
194 MS_EXCEPTION_IF_NULL(new_value_node); in AddFusionTbeClearZeroNode()
195 std::vector<AnfNodePtr> inputs = {new_value_node}; in AddFusionTbeClearZeroNode()
371 auto new_value_node = NewValueNode(clear_zero_prim); in TbeClearZeroNodeFusion() local
372 MS_EXCEPTION_IF_NULL(new_value_node); in TbeClearZeroNodeFusion()
373 std::vector<AnfNodePtr> inputs = {new_value_node}; in TbeClearZeroNodeFusion()
442 auto new_value_node = NewValueNode(clear_zero_prim); in KernelBuildPreprocess() local
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecial_op_eliminate.h508 auto new_value_node = NewValueNode(FillZero(value_node->value())); in OperatorHandle1() local
509 new_value_node->set_has_new_value(value_node->has_new_value()); in OperatorHandle1()
511 return new_value_node; in OperatorHandle1()
522 auto new_value_node = NewValueNode(FillZero(value_node->value())); in OperatorHandle2() local
523 new_value_node->set_has_new_value(value_node->has_new_value()); in OperatorHandle2()
525 return new_value_node; in OperatorHandle2()
590 auto new_value_node = OperatorHandle1(arg, node); in operator() local
591 if (new_value_node != nullptr) { in operator()
592 return new_value_node; in operator()
602 auto new_value_node = OperatorHandle2(arg, node); in operator() local
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_helper.h124 ValueNodePtr new_value_node = std::make_shared<ValueNode>(tensor); in CreateScalarTensorValueNode() local
125 new_value_node->set_abstract(tensor->ToAbstract()); in CreateScalarTensorValueNode()
127 new_value_node->set_kernel_info(kernel_info); in CreateScalarTensorValueNode()
132 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in CreateScalarTensorValueNode()
134 return new_value_node; in CreateScalarTensorValueNode()
Dgraph_kernel_helper.cc139 ValueNodePtr new_value_node = std::make_shared<ValueNode>(scalar_tensor); in ConvertToScalarTensor() local
140 new_value_node->set_abstract(scalar_tensor->ToAbstract()); in ConvertToScalarTensor()
141 new_value_node->set_kernel_info(std::make_shared<device::KernelInfo>()); in ConvertToScalarTensor()
145 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in ConvertToScalarTensor()
147 return new_value_node; in ConvertToScalarTensor()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc147 ValueNodePtr new_value_node = std::make_shared<ValueNode>(value_node->value()); in MakeValueNode() local
148 MS_EXCEPTION_IF_NULL(new_value_node); in MakeValueNode()
149 new_value_node->set_abstract(value_node->abstract()); in MakeValueNode()
150 this->SetKernelInfoForNode(new_value_node); in MakeValueNode()
151 return new_value_node; in MakeValueNode()
590 auto new_value_node = MakeValueNode(value_node)->cast<ValueNodePtr>(); in NewValueNode() local
591 AnfAlgo::SetGraphId(graph_id_, new_value_node.get()); in NewValueNode()
592 return new_value_node; in NewValueNode()
598 ValueNodePtr new_value_node = std::make_shared<ValueNode>(value); in NewValueNode() local
599 MS_EXCEPTION_IF_NULL(new_value_node); in NewValueNode()
[all …]
Dsession_basic.cc298 auto new_value_node = graph->NewValueNode(value_node); in CreateNewValueNode() local
299 graph->FrontBackendlMapAdd(anf, new_value_node); in CreateNewValueNode()
300 graph->AddValueNodeToGraph(new_value_node); in CreateNewValueNode()
301 return new_value_node; in CreateNewValueNode()
708 auto new_value_node = CreateNewValueNode(anf, graph); in GetNewCNodeInputs() local
709 if (new_value_node != nullptr) { in GetNewCNodeInputs()
710 (void)cnode_inputs->emplace_back(new_value_node); in GetNewCNodeInputs()
992 auto new_value_node = CreateValueNodeKernelGraph(attr_input, graph); in CreateValueNode() local
993 if (new_value_node != nullptr) { in CreateValueNode()
994 cnode_inputs.emplace_back(new_value_node); in CreateValueNode()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtopk_split.cc163 auto new_value_node = std::make_shared<ValueNode>(MakeValue(*data)); in Process() local
164 new_cnode->set_input(kTopkIndexK + 1, new_value_node); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc897 ValueNodePtr new_value_node = std::make_shared<ValueNode>(value_node->value()); in MakeValueNode() local
898 MS_EXCEPTION_IF_NULL(new_value_node); in MakeValueNode()
899 new_value_node->set_abstract(value_node->abstract()); in MakeValueNode()
902 new_value_node->set_kernel_info(kernel_info); in MakeValueNode()
915 AnfAlgo::SetSelectKernelBuildInfo(kernel_build_info_builder->Build(), new_value_node.get()); in MakeValueNode()
916 return new_value_node; in MakeValueNode()