Home
last modified time | relevance | path

Searched defs:node_map (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/common/
Dfast_pattern_to_pattern_pass_test.cc473 void Check0(const FuncGraphIndexPtr &fg, const std::map<std::string, AnfNodePtr> &node_map) { in Check0()
499 …stFastMul2 &pass, const FuncGraphIndexPtr &fg, const std::map<std::string, AnfNodePtr> &node_map) { in Check1()
526 void Check2(const CheckPattern &check, const std::map<std::string, AnfNodePtr> &node_map) { in Check2()
584 std::map<std::string, AnfNodePtr> node_map; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/expander/fallback/
Dexpander_fallback.cc50 std::map<AnfNodePtr, AnfNodePtr> node_map; in DumpGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dquant_strategy.cc135 auto node_map = manager->node_users(); in CheckAscendSpec() local
Dweight_quantizer.cc205 auto node_map = manager->node_users(); in LinearQuant() local
Dinsert_quant_node_manager.cc929 auto node_map = manager->node_users(); in InsertAscendAntiQuantNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/expander/bprop/
Dbprop.cc269 std::vector<NodePtr> node_map(input_nodes); in Build() local
309 std::unordered_map<NodePtr, size_t> node_map; in BuildBpropOpGraph() local
444 std::map<AnfNodePtr, AnfNodePtr> node_map; in DumpResult() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dgraph_kernel_utils.cc236 std::map<inner::NodePtr, AnfNodePtr> node_map; in LiteGraph2AnfGraph() local
344 std::map<AnfNodePtr, inner::NodePtr> node_map; in AnfGraph2LiteGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dgraph_kernel_splitter_with_py.cc147 std::map<AnfNodePtr, AnfNodePtr> node_map; in DecodeSplitNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/
Dtop_cell.h53 mindspore::HashMap<std::string, std::pair<AnfNodePtr, std::vector<int64_t>>> node_map; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dgru_fusion_pass.cc577 bool CheckClosure(const std::map<uint32_t, uint32_t> &node_map) { in CheckClosure() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/
Dcontrol_node_scheduler.cc192 auto &node_map = control_node_ds_actor->data_node_position_map_; in BuildDataSourceActorForControlNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/express_ir/
Donnx_exporter.cc1375 std::map<AnfNodePtr, std::string> node_map; in GetOnnxProtoString() local