Searched refs:fusion_scope_info (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | fusion_tbe_json_creator.cc | 33 bool FusionBuildTbeJsonCreator::GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *f… in GenJson() argument
36 MS_LOG(DEBUG) << "Start Generate Fusion Json, Fusion Node: " << fusion_scope_info.full_name; in GenJson()
41 if (!GenOpListJson(fusion_scope_info, &op_list_json)) { in GenJson()
51 (*fusion_json)[kJScopeID] = fusion_scope_info.scope_id; in GenJson()
52 (*fusion_json)[kJFullName] = fusion_scope_info.full_name; in GenJson()
57 bool FusionBuildTbeJsonCreator::GenOpListJson(const FusionScopeInfo &fusion_scope_info, in GenOpListJson() argument
61 if (!CheckInput(fusion_scope_info)) { in GenOpListJson()
62 for (const auto &cnode : fusion_scope_info.compute_nodes) { in GenOpListJson()
63 … MS_LOG(WARNING) << "Fusion Error: check input failed, scope id: " << fusion_scope_info.scope_id in GenOpListJson()
70 auto compute_nodes = fusion_scope_info.compute_nodes; in GenOpListJson()
[all …]
|
| D | fusion_tbe_json_creator.h | 28 bool GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *fusion_json) override;
31 …bool GenOpListJson(const FusionScopeInfo &fusion_scope_info, std::vector<nlohmann::json> *fusion_j…
46 bool CheckInput(const FusionScopeInfo &fusion_scope_info);
|
| D | tbe_json_creator.h | 70 …virtual bool GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *fusion_json) { retu… in GenJson() argument
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 305 FusionScopeInfo fusion_scope_info(0, full_name, input_nodes, compute_nodes, {}); in TEST_F() local
308 EXPECT_TRUE(tbe_json_creator->GenJson(fusion_scope_info, &fusion_json)); in TEST_F()
364 FusionScopeInfo fusion_scope_info(0, full_name, input_nodes, compute_nodes, {}); in TEST_F() local
367 EXPECT_TRUE(tbe_json_creator->GenJson(fusion_scope_info, &fusion_json)); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.cc | 385 bool TbeAdapter::GetSpecDataInput(const FusionScopeInfo &fusion_scope_info, in GetSpecDataInput() argument
388 auto input_nodes = fusion_scope_info.input_nodes; in GetSpecDataInput()
389 auto compute_nodes = fusion_scope_info.compute_nodes; in GetSpecDataInput()
|
| D | tbe_adapter.h | 142 static bool GetSpecDataInput(const FusionScopeInfo &fusion_scope_info,
|