Home
last modified time | relevance | path

Searched defs:AnfNodePtr (Results 1 – 25 of 67) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/debug/
Ddump_proto.cc350 const std::map<AnfNodePtr, size_t> &apply_map, in GetOpNodeInputId()
430 std::map<AnfNodePtr, size_t> *const_map_ptr) { in ExportCNodes()
452 std::map<AnfNodePtr, size_t> *apply_map_ptr, in ExportCNode()
492 const std::map<AnfNodePtr, size_t> &apply_map, in ExportFuncGraphOutput()
515 static bool CompareValue(const std::pair<AnfNodePtr, size_t> &x, const std::pair<AnfNodePtr, size_t… in CompareValue()
519 void ProtoExporter::ExportValueNodes(const std::map<AnfNodePtr, size_t> &const_map, irpb::GraphProt… in ExportValueNodes()
522 [](const std::pair<AnfNodePtr, size_t> &item) { return item; }); in ExportValueNodes()
Danf_ir_dump.cc177 int32_t DumpParams(const FuncGraphPtr &graph, std::ostringstream &buffer, OrderedMap<AnfNodePtr, in… in DumpParams()
248 void DumpOperands(const AnfNodePtr &nd, OrderedMap<AnfNodePtr, int32_t> *para_map, in DumpOperands()
405 void DumpCNode(const CNodePtr &nd, const FuncGraphPtr &sub_graph, OrderedMap<AnfNodePtr, int32_t> *… in DumpCNode()
482 void DumpIRInSubgraph(const std::vector<AnfNodePtr> &nodes, OrderedMap<AnfNodePtr, int32_t> *para_m… in DumpIRInSubgraph()
521 … const FuncGraphPtr &graph, OrderedMap<AnfNodePtr, int32_t> *para_map, std::ofstream &fout) { in DumpSubgraph()
Danf_ir_utils.cc390 const std::map<AnfNodePtr, int> &apply_map) { in GetAnfNodeText()
422 … OrderedMap<AnfNodePtr, int, ParamPtrHasher, ParamPtrEqual> *param_map) { in OutputParameters()
488 std::map<AnfNodePtr, int> *const apply_map) { in OutputCNodeText()
517 std::map<AnfNodePtr, int> *const apply_map) { in OutputCNode()
/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Dproto_exporter.cc286 const std::map<AnfNodePtr, size_t> &apply_map, in GetOpNodeInputId()
391 … std::map<AnfNodePtr, size_t> *const_map_ptr, LocDebugDumpMode dump_location) { in ExportCNodes()
413 std::map<AnfNodePtr, size_t> *apply_map_ptr, in ExportCNode()
465 const std::map<AnfNodePtr, size_t> &apply_map, in ExportFuncGraphOutput()
484 static bool CompareValue(const std::pair<AnfNodePtr, size_t> &x, const std::pair<AnfNodePtr, size_t… in CompareValue()
488 void DebuggerProtoExporter::ExportValueNodes(const std::map<AnfNodePtr, size_t> &const_map, in ExportValueNodes()
492 [](const std::pair<AnfNodePtr, size_t> &item) { return item; }); in ExportValueNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dparallel_fusion.cc34 OrderedMap<AnfNodePtr, NodeRelation> *node_rels) { in ProcessThroughPassCNode()
84 void ProcessTailMakeTupleCNode(OrderedMap<AnfNodePtr, NodeRelation> *node_rels) { in ProcessTailMakeTupleCNode()
129 bool IsSingleInputNode(const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const AnfNodePtr &nod… in IsSingleInputNode()
136 bool IsSingleOutputNode(const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const AnfNodePtr &no… in IsSingleOutputNode()
143 bool IsMultiInputsNode(const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const AnfNodePtr &nod… in IsMultiInputsNode()
150 bool IsMultiOutputsNode(const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const AnfNodePtr &no… in IsMultiOutputsNode()
157 bool IsNoInputsNode(const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const AnfNodePtr &node) { in IsNoInputsNode()
164 bool IsNoOutputsNode(const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const AnfNodePtr &node)… in IsNoOutputsNode()
171 void ProcessLocalStructure(OrderedMap<AnfNodePtr, NodeRelation> *node_rels, std::set<AnfNodePtr> *v… in ProcessLocalStructure()
228 …const OrderedMap<AnfNodePtr, NodeRelation> &node_rels, const std::set<AnfNodePtr> &virtual_noout_n… in GetInterestNodeIds()
[all …]
Dshape_ops_splitter.cc56 …std::for_each(index_set.cbegin(), index_set.cend(), [&users_info](const std::pair<AnfNodePtr, int>… in SplitNode()
81 [](const std::pair<AnfNodePtr, int> &iter) { return iter.first; }); in IsMultiUserShapeOps()
Dadd_stitch_atomic_clean_gpu.cc104 [](const std::pair<AnfNodePtr, int> &pair) { return pair; }); in FindInnerCNodeUsers()
/third_party/mindspore/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc602 std::unordered_map<AnfNodePtr, OpMergedInfo> *op_merged_infos_ptr) { in MatchAndMark()
666 void OnnxExporter::ExportNodes(const FuncGraphPtr &func_graph, std::map<AnfNodePtr, size_t> *node_m… in ExportNodes()
742 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *const graph_proto) { in ExportPrimReshape()
775 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *const graph_proto) { in ExportPrimReduce()
815 std::map<AnfNodePtr, size_t> *node_map_ptr, in ExportPrimTranspose()
851 std::map<AnfNodePtr, size_t> *node_map_ptr, in ExportPrimStridedSlice()
948 std::map<AnfNodePtr, size_t> *node_map_ptr, in ExportPrimResizeNearestNeighbor()
1009 std::map<AnfNodePtr, size_t> *node_map_ptr, in ExportPrimExpandDims()
1055 std::map<AnfNodePtr, size_t> *node_map_ptr, in ExportPrimBatchMatMul()
1169 … std::map<AnfNodePtr, size_t> *node_map_ptr, onnx::GraphProto *const graph_proto) { in ExportPrimGeLU()
[all …]
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dspecial_op_eliminate.h103 void Visit(const AnfNodePtr &) override {} in Visit()
120 void Visit(const AnfNodePtr &) override {} in Visit()
138 void Visit(const AnfNodePtr &) override {} in Visit()
195 void Visit(const AnfNodePtr &) override {} in Visit()
252 void Visit(const AnfNodePtr &) override {} in Visit()
271 void Visit(const AnfNodePtr &) override {} in Visit()
310 void Visit(const AnfNodePtr &) override {} in Visit()
349 void Visit(const AnfNodePtr &) override {} in Visit()
Dinline.h96 bool IsForceInline(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &) { in IsForceInline()
365 bool IsUniqueUse(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &) { in IsUniqueUse()
373 bool IsTrivial(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &) { in IsTrivial()
384 bool IsCore(InlinerBase *, const FuncGraphPtr &fg, const AnfNodePtr &) { return fg->has_flag("core"… in IsCore()
397 bool IsNotRecursive(InlinerBase *inliner, const FuncGraphPtr &fg, const AnfNodePtr &) { in IsNotRecursive()
Denv_item_eliminate.h206 void Visit(const AnfNodePtr &) override { is_match_ = true; } in Visit()
276 void Visit(const AnfNodePtr &) override { is_match_ = true; } in Visit()
346 void Visit(const AnfNodePtr &) override { is_match_ = true; } in Visit()
408 void Visit(const AnfNodePtr &) override { is_match_ = true; } in Visit()
481 void Visit(const AnfNodePtr &) override { is_match_ = true; } in Visit()
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dmap.cc76 …std::any_of(arg_pairs.begin(), arg_pairs.end(), [&num, size](const std::pair<AnfNodePtr, TypePtr> … in FullMakeList()
112 [&func_graph, pos](const std::pair<AnfNodePtr, Any> &item) { in FullMakeList()
135 …std::any_of(arg_pairs.begin(), arg_pairs.end(), [&num, size](const std::pair<AnfNodePtr, TypePtr> … in FullMakeTuple()
171 [&func_graph, &pos](const std::pair<AnfNodePtr, Any> &item) { in FullMakeTuple()
248 …std::any_of(arg_pairs.begin(), arg_pairs.end(), [pair](const std::pair<AnfNodePtr, TypePtr> &item)… in Make()
Dcomposite.cc107 [](const std::pair<AnfNodePtr, Any> &item) { return item.first; }); in FullMake()
119 …std::any_of(arg_map.begin(), arg_map.end(), [&num, size](const std::pair<AnfNodePtr, TypePtr> &ite… in FullMake()
154 [&func_graph, pos](const std::pair<AnfNodePtr, Any> &item) { in FullMake()
177 …std::any_of(arg_map.begin(), arg_map.end(), [&num, size](const std::pair<AnfNodePtr, TypePtr> &ite… in FullMake()
212 [&func_graph, &pos](std::pair<AnfNodePtr, Any> item) { in FullMake()
283 …_not_same = std::any_of(arg_map.begin(), arg_map.end(), [pair](const std::pair<AnfNodePtr, TypePtr… in Make()
323 …[type_tensor](const std::pair<AnfNodePtr, TypePtr> &item) { return IsSubType(item.second, type_ten… in Harmonize()
796 [fg_ptr](AnfNodePtr) { return fg_ptr->add_parameter(); }); in MakeCond() argument
831 [fg_ptr](AnfNodePtr) { return fg_ptr->add_parameter(); }); in MakeNext() argument
/third_party/mindspore/mindspore/core/ir/
Dfunc_graph_extends.cc87 std::unordered_map<AnfNodePtr, AnfNodePtr> *repl_nodes) const { in GenerateVarParams()
123 std::unordered_map<AnfNodePtr, AnfNodePtr> *repl_nodes) const { in GenerateKwParams()
179 … std::unordered_map<AnfNodePtr, AnfNodePtr> *repl_nodes) const { in GenerateKwargReplNode()
206 std::unordered_map<AnfNodePtr, AnfNodePtr> *repl_nodes) const { in GenerateDefaultValue()
Dpattern_matcher.h67 T GetNode(const AnfNodePtr &) const { in GetNode()
/third_party/mindspore/tests/ut/cpp/stub/ge/
Dge_task_launch_stub.cc34 bool HcclAdapter::GenTask(const AnfNodePtr &, HcclDataType, std::vector<HcclTaskInfo> *) const { re… in GenTask() argument
35 int64_t HcclAdapter::CalcWorkspaceSize(const AnfNodePtr &, HcclDataType) const { return 0; } in CalcWorkspaceSize() argument
37 std::string HcclAdapter::GetHcclType(const AnfNodePtr &) { return ""; } in GetHcclType()
/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc57 void CalcNodeRefCount(const FuncGraphPtr &graph, std::map<AnfNodePtr, size_t> *nodes_ref) { in CalcNodeRefCount()
139 std::vector<AnfNodePtr> GetNextNodes(const AnfNodePtr &node, std::map<AnfNodePtr, size_t> *nodes_re… in GetNextNodes()
322 … std::map<AnfNodePtr, std::string> *node_input_target_map) { in ParallelSortDecideNextHandleTarget()
411 void AddSegmentDependency(const FuncGraphPtr &graph, const std::map<AnfNodePtr, GraphSegmentPtr> &n… in AddSegmentDependency()
504 std::map<AnfNodePtr, GraphSegmentPtr> *node_to_segment) { in AddSegment()
554 std::map<AnfNodePtr, GraphSegmentPtr> *node_to_segment, in SplitDynamicNodeSegment()
594 std::map<AnfNodePtr, GraphSegmentPtr> *node_to_segment) { in NodesToSegments()
Dsegment_runner.cc62 … [&seen](const std::pair<AnfNodePtr, int64_t> &u) -> bool { in GetOutput()
143 … [](const std::pair<AnfNodePtr, AnfNodePtr> &elem) -> AnfNodePtr { return elem.first; }); in TransformSegmentToAnfGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/rts/
Drt_kernel.cc43 bool RtKernel::Init(const mindspore::AnfNodePtr & /*anf_node*/) { return true; } in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc564 std::map<AnfNodePtr, nlohmann::json> *node_json_map, in GenStitchJson()
656 std::map<AnfNodePtr, nlohmann::json> *node_json_map) { in GenSingleJsons()
680 std::map<AnfNodePtr, nlohmann::json> *node_json_map) { in UpdateTensorName()
710 … const std::map<AnfNodePtr, nlohmann::json> &node_json_map) { in CreateInputsJson()
735 … const std::map<AnfNodePtr, nlohmann::json> &node_json_map, in GenParallelJson()
800 … const std::map<AnfNodePtr, nlohmann::json> &node_json_map) { in CreateOutputsJson()
/third_party/mindspore/mindspore/lite/tools/common/
Dfunc_graph_subgraph.cc274 [this](const std::pair<AnfNodePtr, int> &output_pair) { in InitSubGraphOutNode()
478 std::map<AnfNodePtr, AnfNodePtr> *partial_inputs_and_subgraph_input_map) { in CreateParameterForPartialSubGraph()
545 …const FuncGraphPtr &sub_graph, const std::map<AnfNodePtr, AnfNodePtr> &partial_inputs_and_subgraph… in CreateCNodeForPartialSubGraph()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.cc294 std::map<const AnfNodePtr, FusionDataType> *spec_data_input) { in GetSpecInputLayers()
313 … const std::map<const AnfNodePtr, tbe::FusionDataType> &spec_data_input) { in FusionDescJsonPass()
386 … std::map<const AnfNodePtr, tbe::FusionDataType> *spec_data_input) { in GetSpecDataInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dnode_pass.cc32 void AddOutputAndCallerToMap(const CNodePtr &cnode, std::unordered_map<AnfNodePtr, AnfNodePtr> *out… in AddOutputAndCallerToMap()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dconvert_cast_format.cc113 const std::shared_ptr<std::vector<std::pair<AnfNodePtr, int>>> &used_cast_node_list, in CalculateFormat()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc452 std::map<AnfNodePtr, size_t> *parameter_index, in GetParamAndOutputIndex()
461 const std::map<AnfNodePtr, size_t> &parameter_index, in GetSingleOpInputTensors()
470 … const std::map<AnfNodePtr, size_t> &parameter_index, in GetSingleOpInputTensorByIndex()

123