Home
last modified time | relevance | path

Searched refs:IsValueNode (Results 1 – 25 of 112) sorted by relevance

12345

/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dsymbol_resolver.cc43 … if (!IsValueNode<parse::NameSpace>(namespace_node) || !IsValueNode<parse::Symbol>(symbol_node)) { in operator ()()
73 attr_node.CheckFunc(IsValueNode<StringImm>, node)); in operator ()()
77 …ns_node.CheckFunc(IsValueNode<parse::NameSpace>, node) && attr_node.CheckFunc(IsValueNode<StringIm… in operator ()()
81 …bool_node.CheckFunc(IsValueNode<BoolImm>, node) && attr_node.CheckFunc(IsValueNode<StringImm>, nod… in operator ()()
85 …ns_node.CheckFunc(IsValueNode<parse::NameSpace>, node) && sym_node.CheckFunc(IsValueNode<parse::Sy… in operator ()()
Denv_item_eliminate.h67 if (!IsValueNode<SymbolicKeyInstance>(inputs[2])) { in operator()
115 if (!IsValueNode<FuncGraph>(output_outer)) { in operator()
131 if (!IsValueNode<SymbolicKeyInstance>(inputs[2])) { in operator()
165 (IsValueNode<EnvInstance>(c1.GetNode(node)) && IsVNode(c2.GetNode(node)) && in operator()
229 return IsValueNode<SymbolicKeyInstance>(inputs[2]); in operator()
231 …AnfVisitor::Match(prim::kPrimEnvGetItem, {IsSetCNode, IsValueNode<SymbolicKeyInstance>, IsNode})(n… in operator()
260 if (!IsValueNode<SymbolicKeyInstance>(inputs[2])) { in operator()
314 return IsValueNode<FuncGraph>(cnode->input(0)); in operator()
316 …AnfVisitor::Match(prim::kPrimEnvGetItem, {IsGCNode, IsValueNode<SymbolicKeyInstance>, IsNode})(nod… in operator()
370 …AnfVisitor::Match(prim::kPrimEnvGetItem, {IsSwNode, IsValueNode<SymbolicKeyInstance>, IsNode})(nod… in operator()
[all …]
Dspecialize_transform.h101 if (!IsValueNode<FuncGraph>(inputs[0])) { in operator()
113 if (IsValueNode<FuncGraph>(inputs[i]) || IsValueNode<Primitive>(inputs[i]) || in operator()
114 IsValueNode<tensor::Tensor>(inputs[i]) || IsValueNode<parse::NameSpace>(inputs[i])) { in operator()
Ditem_tuple_or_list_eliminate.h70 if (sequeue_ != nullptr && IsValueNode<Int64Imm>(vnode)) { in Visit()
119 if (tuple_ != nullptr && IsValueNode<Int64Imm>(vnode)) { in Visit()
164 if (IsValueNode<ValueTuple>(vnode)) { in Visit()
168 if (tuple_ != nullptr && IsValueNode<Int64Imm>(vnode)) { in Visit()
247 if (args_.empty() && IsValueNode<ValueTuple>(vnode)) { in Visit()
257 } else if (args_.empty() && IsValueNode<ValueList>(vnode)) { in Visit()
267 } else if (!args_.empty() && IsValueNode<Int64Imm>(vnode)) { in Visit()
329 if (tuple_ != nullptr && IsValueNode<Int64Imm>(vnode)) { in Visit()
369 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
370 AnfVisitor::Match(prim::kPrimListGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
Dgradient_eliminate.cc41 if (IsValueNode<Primitive>(value_node)) { in CheckIfEmbedJ()
87 if (IsValueNode<FuncGraph>(vnode)) { in ExpandJ()
98 if (IsValueNode<Primitive>(vnode)) { in ExpandJ()
Dreduce_eliminate.h96 if (IsValueNode<tensor::Tensor>(node)) { in Visit()
123 if (IsValueNode<Int64Imm>(vnode)) { in Visit()
137 if (IsValueNode<ValueTuple>(vnode)) { in Visit()
Dcast_eliminate.cc36 bool IsValueNode(const AnfNodePtr &node) { return IsVNode(TransThroughDepend(node)); } in IsValueNode() function
41 AnfVisitor::Match(prim::kPrimCast, {IsNode, IsValueNode})(node); in operator ()()
Dcall_graph_tuple_transform.h112 if (IsValueNode<FuncGraph>(node)) { in TransformBranchNode()
123 if (IsValueNode<FuncGraph>(partial_inputs[1])) { in TransformBranchNode()
191 if (!IsValueNode<FuncGraph>(tuple_inputs[i])) { in TransformLayerNode()
Dincorporate_call.h132 if (cnode->size() < 1 || !IsValueNode<FuncGraph>(cnode->input(0))) { in Visit()
177 …AnfVisitor::Match(prim::kPrimSwitch, {IsNode, IsValueNode<FuncGraph>, IsValueNode<FuncGraph>})(inp… in operator()
Dincorporate_getitem.h100 if (!IsValueNode<FuncGraph>(output_outer)) { in operator()
331 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
494 if (cnode->size() == 0 || !IsValueNode<FuncGraph>(cnode->input(0))) { in Visit()
535 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
574 if (!IsValueNode<FuncGraph>(fg_cnode->input(0))) { in Visit()
621 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
630 …AnfVisitor::Match(prim::kPrimSwitch, {IsNode, IsValueNode<FuncGraph>, IsValueNode<FuncGraph>})(swi… in operator()
883 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
931 if (IsPrimitiveCNode(cnode, prim::kPrimMakeTuple) && IsValueNode<FuncGraph>(inputs[1])) { in Visit()
971 AnfVisitor::Match(prim::kPrimTupleGetItem, {IsCNode, IsValueNode<Int64Imm>})(node); in operator()
[all …]
Dgrad_var_prepare.cc64 if (IsValueNode<prim::DoSignaturePrimitive>(node)) { in GetMetaFuncGraphOfValueNode()
129 if (!IsValueNode<FuncGraph>(func_node)) { in operator ()()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/
Dnode_info.cc68 (IsValueNode<ValueList>(node_inputs[1]) || IsValueNode<ValueTuple>(node_inputs[1]))) { in ExtractInputParameterByNode()
70 if (IsValueNode<ValueList>(node_inputs[1])) { in ExtractInputParameterByNode()
89 …} else if (input->isa<CNode>() || IsValueNode<tensor::Tensor>(input) || IsValueNode<RefKey>(input)… in ExtractInputParameterByNode()
136 if (!input->isa<CNode>() && !input->isa<Parameter>() && !IsValueNode<tensor::Tensor>(input)) { in GetInputsTypeLen()
158 (IsValueNode<ValueList>(node_inputs[1]) || IsValueNode<ValueTuple>(node_inputs[1]))) { in ExtractInputTypeLengthByNode()
160 if (IsValueNode<ValueList>(node_inputs[1])) { in ExtractInputTypeLengthByNode()
183 if (IsValueNode<RefKey>(input)) { in ExtractInputTypeLengthByNode()
191 … } else if (input->isa<CNode>() || input->isa<Parameter>() || IsValueNode<tensor::Tensor>(input)) { in ExtractInputTypeLengthByNode()
233 if (!IsValueNode<RefKey>(node)) { in FindParameterByRefKeyNode()
268 if ((cnode == nullptr) || !IsValueNode<Primitive>(cnode->input(0))) { in AnfNodeIsPrimitive()
[all …]
/third_party/mindspore/mindspore/ccsrc/debug/
Ddraw.cc134 if (IsValueNode<FuncGraph>(t)) { in DrawEdges()
243 if (IsValueNode<FuncGraph>(node)) { in Shape()
263 if (IsValueNode<FuncGraph>(node)) { in Color()
281 if (node->isa<CNode>() || (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node))) { in Head()
401 if (IsValueNode<MetaFuncGraph>(node)) { in DrawValueNode()
403 } else if (IsValueNode<parse::NameSpace>(node)) { in DrawValueNode()
405 } else if (IsValueNode<parse::Symbol>(node)) { in DrawValueNode()
492 if (IsValueNode<Primitive>(node->input(0)) && is_modelgraph) { in DrawCNode()
536 } else if (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node)) { in Node()
540 if (IsValueNode<FuncGraph>(node)) { in Node()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dstep_parallel_utils.cc86 if (IsValueNode<ValueList>(node)) { in GetValueListShape()
88 } else if (IsValueNode<ValueTuple>(node)) { in GetValueListShape()
108 if (IsValueNode<ValueList>(node) || IsValueNode<ValueTuple>(node)) { in GetNodeShape()
114 if (IsValueNode<Primitive>(cnode->input(0))) { in GetNodeShape()
154 if (!IsValueNode<Primitive>(node->input(0))) { in CreateInstanceName()
Dstep_parallel.cc275 if (!IsValueNode<Primitive>(uses_cnode->input(0))) { in ForwardCommunication()
498 if ((cnode == nullptr) || !IsValueNode<Primitive>(cnode->input(0))) { in HasStrategy()
525 if (!IsValueNode<Primitive>(cnode->input(0))) { in FindCommunicationOp()
558 if (IsValueNode<Primitive>(node->input(0))) { in StepRedistribution()
571 if (!IsValueNode<Primitive>(use_cnode->input(0))) { in StepRedistribution()
648 if (IsValueNode<ValueList>(node)) { in SplitTensorList()
694 if (use_cnode == nullptr || !IsValueNode<Primitive>(use_cnode->input(0))) { in StepSplitTensor()
706 if (IsValueNode<ValueList>(node) || IsValueNode<ValueTuple>(node)) { in StepSplitTensor()
913 if (IsValueNode<RefKey>(node)) { in FindParameterByValueNode()
952 if (!IsValueNode<Primitive>(cnode->input(0))) { in FindParameter()
[all …]
Dparameter_manager.cc58 …if ((cnode_ptr == nullptr) || !IsValueNode<Primitive>(cnode_ptr->input(0)) || !IsCareNode(cnode_pt… in FindRefKeyNodeUsers()
74 if ((c == nullptr) || !IsValueNode<Primitive>(c->input(0)) || !IsCareNode(c)) { in FindRefKeyNodeUsers()
91 if ((c == nullptr) || !IsValueNode<Primitive>(c->input(0)) || !IsCareNode(c)) { in FindRefKeyNodeUsers()
139 if (IsValueNode<RefKey>(one_input)) { in CNodeWithRefKeys()
178 if (IsValueNode<FuncGraph>(use_node->input(0))) { in IsUsedParameter()
186 if (!IsSomePrimitive(cnode, J) || !IsValueNode<FuncGraph>(cnode->input(1))) { in IsUsedParameter()
308 if (IsValueNode<SymbolicKeyInstance>(node)) { in HandleSymbolicKeyInstance()
/third_party/mindspore/tests/ut/cpp/ir/
Danf_test.cc39 ASSERT_EQ(IsValueNode<Primitive>(c), true); in TEST_F()
40 ASSERT_EQ(IsValueNode<FuncGraph>(c), false); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dirpass.cc114 …MakeSubstitution(std::make_shared<ResetDeferInline>(), "reset_defer_inline", IsValueNode<FuncGraph… in OptimizeIRPassLib()
149 IsValueNode<RefKey>, opt::FORCE_RENORM); in OptimizeIRPassLib()
184 …MakeSubstitution(std::make_shared<ReplaceApplicator>(), "replace_applicator", IsValueNode<FuncGrap… in OptimizeIRPassLib()
263 … "set_cell_output_no_recompute", IsValueNode<FuncGraph>); in OptimizeIRPassLib()
Dopt.cc102 if (IsValueNode<FuncGraph>(node) || IsValueNode<RefKey>(node)) { in isTraversable()
132 if (IsValueNode<FuncGraph>(node)) { in UpdateTransformingListForSubstitutions()
148 if (IsValueNode<FuncGraph>(node)) { in UpdateTransformingListForIR()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc126 if (IsValueNode<FuncGraph>(cnode->input(0))) { in LabelParameterStart()
206 if (!IsValueNode<FuncGraph>(node)) { in Coloring()
249 if (IsValueNode<FuncGraph>(user_node->input(0))) { in BroadCastColoring()
292 if (IsValueNode<FuncGraph>(cnode->input(0))) { in CreateOpInfo()
367 if (IsValueNode<FuncGraph>(temp_cnode->input(0))) { in GetActualOpUsers()
423 if (IsValueNode<FuncGraph>(cnode->input(0))) { in HandleSharedParameter()
467 if (IsValueNode<FuncGraph>(node->input(0))) { in ParameterColoring()
492 if (cnode != nullptr && IsValueNode<FuncGraph>(cnode->input(0))) { in GetShapeType()
520 if (IsValueNode<FuncGraph>(cnode->input(0))) { in FindPipelineCareNode()
674 if (IsValueNode<FuncGraph>(cnode->input(0))) { in InsertReceive()
[all …]
/third_party/mindspore/mindspore/ccsrc/vm/
Dsegment_runner.cc81 if (node->isa<ValueNode>() && !IsValueNode<FuncGraph>(node)) { in RefSubGraphNode()
117 if (!IsValueNode<Primitive>(inps[0]) && in TransformSegmentToAnfGraph()
118 !(IsValueNode<FuncGraph>(inps[0]) && in TransformSegmentToAnfGraph()
/third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/
Dacl_vm.cc126 if (IsValueNode<FuncGraph>(node)) { in Ref()
131 if (IsValueNode<Primitive>(node)) { in Ref()
133 } else if (IsValueNode<tensor::Tensor>(node)) { in Ref()
224 if (!IsValueNode<FuncGraph>(fn)) { in AddPartial()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dvalidator.cc45 if (!IsValueNode<Primitive>(node)) { in ValidateOperation()
84 if (!IsValueNode<StringImm>(node)) { in CheckAbstractScalar()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dconvert_const_input_to_tensor_input.cc78 if (IsValueNode<Scalar>(input_node) || IsValueNode<ValueTuple>(input_node)) { in ConstInputToTensorInput()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dcontrol_node_parser.cc71 if (IsValueNode<FuncGraph>(partial_node)) { in FetchParameterBySwitchNode()
120 } else if (tuple_inputs[i]->isa<ValueNode>() && IsValueNode<FuncGraph>(tuple_inputs[i])) { in FetchParameterBySwitchLayerNode()
429 } else if (cnode->input(0)->isa<CNode>() || IsValueNode<FuncGraph>(cnode->input(0))) { in FetchParameterByControlNode()
461 } else if (IsValueNode<FuncGraph>(func_graph->output())) { in FetchFuncGraphInNode()
467 } else if (IsValueNode<FuncGraph>(node)) { in FetchFuncGraphInNode()
473 } else if (IsValueNode<FuncGraph>(func_graph->output())) { in FetchFuncGraphInNode()
520 …return inputs[0]->isa<CNode>() || (inputs[0]->isa<ValueNode>() && IsValueNode<FuncGraph>(inputs[0]… in IsCallNode()
545 IsValueNode<FuncGraph>(tuple_inputs[kMakeTupleInputStartPos])) { in IsSubCallNode()
551 (output->isa<ValueNode>() && IsValueNode<FuncGraph>(output)); in IsSubCallNode()
590 } else if (IsValueNode<FuncGraph>(cnode_inputs[i])) { in FetchFuncGraphbyCallNode()
[all …]

12345