Searched refs:EraseNodeAttr (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | erase_visit_attr.cc | 32 AnfAlgo::EraseNodeAttr(kAttrVisited, node); in Process()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | add_stitch_atomic_clean_gpu.cc | 41 AnfAlgo::EraseNodeAttr(kAttrStitch, atomic_add_node_); in CreateInplaceAssignNode()
|
| /third_party/mindspore/tests/ut/cpp/session/ |
| D | anf_runtime_algorithm_test.cc | 192 TEST_F(AnfRuntimeAlgorithmTest, EraseNodeAttr) { in TEST_F() argument
199 AnfAlgo::EraseNodeAttr("test_set_attr_v2", add); in TEST_F()
201 EXPECT_THROW(AnfAlgo::EraseNodeAttr("test_set_attr_v2", nullptr), std::runtime_error); in TEST_F()
204 EXPECT_THROW(AnfAlgo::EraseNodeAttr("test_set_attr_v2", parameter), std::runtime_error); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 129 static void EraseNodeAttr(const std::string &key, AnfNodePtr node);
|
| D | ascend_session.cc | 1188 AnfAlgo::EraseNodeAttr(kAttrPynativeNextOpName, cnode); in SelectKernel()
1189 AnfAlgo::EraseNodeAttr(kAttrPynativeNextIndex, cnode); in SelectKernel()
|
| D | ascend_auto_monad.cc | 1619 AnfAlgo::EraseNodeAttr(kAttrChildGraph, cnode); in GenerateExecuteOrder()
|
| D | anf_runtime_algorithm.cc | 540 void AnfRuntimeAlgorithm::EraseNodeAttr(const std::string &key, const AnfNodePtr node) { in EraseNodeAttr() function in mindspore::session::AnfRuntimeAlgorithm
|
| /third_party/mindspore/tests/ut/cpp/pre_activate/ascend/ir_fusion/ |
| D | layer_norm_beta_gamma_backprop_fusion_test.cc | 220 AnfAlgo::EraseNodeAttr("shape_gamma", layer); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 540 AnfAlgo::EraseNodeAttr(kAttrFusionId, node); in Run()
|