| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 78 EXPECT_TRUE(tbe_json_creator_select->GenJson(relu1, &kernel_json)); in TEST_F()
80 EXPECT_TRUE(tbe_json_creator_check->GenJson(relu1, &kernel_json)); in TEST_F()
82 EXPECT_TRUE(tbe_json_creator_build->GenJson(relu1, &kernel_json)); in TEST_F()
121 EXPECT_TRUE(tbe_json_creator_select->GenJson(conv2d_backprop_filter, &kernel_json)); in TEST_F()
123 EXPECT_TRUE(tbe_json_creator_check->GenJson(conv2d_backprop_filter, &kernel_json)); in TEST_F()
125 EXPECT_TRUE(tbe_json_creator_build->GenJson(conv2d_backprop_filter, &kernel_json)); in TEST_F()
179 EXPECT_TRUE(tbe_json_creator_select->GenJson(dynamic_rnn, &kernel_json)); in TEST_F()
181 EXPECT_TRUE(tbe_json_creator_check->GenJson(dynamic_rnn, &kernel_json)); in TEST_F()
183 EXPECT_TRUE(tbe_json_creator_build->GenJson(dynamic_rnn, &kernel_json)); in TEST_F()
233 EXPECT_TRUE(tbe_json_creator_select->GenJson(layer_norm, &kernel_json)); in TEST_F()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_creator.h | 69 virtual bool GenJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) { return false; } in GenJson() function
70 …virtual bool GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *fusion_json) { retu… in GenJson() function
|
| D | fusion_tbe_json_creator.h | 28 bool GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *fusion_json) override;
|
| D | single_tbe_json_creator.h | 27 bool GenJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) override;
|
| D | fusion_tbe_json_creator.cc | 33 bool FusionBuildTbeJsonCreator::GenJson(const FusionScopeInfo &fusion_scope_info, nlohmann::json *f… in GenJson() function in mindspore::kernel::FusionBuildTbeJsonCreator
|
| D | single_tbe_json_creator.cc | 31 bool SingleTbeJsonCreator::GenJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) { in GenJson() function in mindspore::kernel::SingleTbeJsonCreator
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | ascend_kernel_compile.cc | 552 if (!json_creator->GenJson(node, &kernel_json)) { in AscendPreBuild()
599 if (!json_creator->GenJson(node, &kernel_json)) { in AscendSingleOpCompile()
666 if (!json_creator->GenJson(fusion_scope_iter, &fusion_op)) { in AscendFusionOpCompile()
804 if (!json_creator->GenJson(node, &kernel_info)) { in AscendOpSelectFormat()
822 if (!json_creator->GenJson(node, &kernel_info)) { in AscendOpCheckSupported()
|
| D | tbe_utils.cc | 388 if (!json_creator->GenJson(node, &kernel_json)) { in GetCompileInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_helper.cc | 100 bool GenJson(const AnfNodePtrList &op_nodes, const std::pair<AnfNodePtrList, AnfNodePtrList> &in_an… in GenJson() function
519 return GenJson(op_nodes, in_and_out, dump_option, op_desc, address_node_map); in AnfToJsonDesc()
547 return GenJson(op_nodes, in_and_out, dump_option, op_desc); in AnfToJsonDesc()
|