| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 106 bool PyExpander::ExpandJsonInfo(const AnfNodePtr &node, nlohmann::json *kernel_json) { in ExpandJsonInfo()
114 nlohmann::json kernel_json; in CreateExpandFuncGraph() local
257 bool ComplexOpExpander::ExpandJsonInfo(const AnfNodePtr &node, nlohmann::json *kernel_json) { in ExpandJsonInfo()
|
| D | graph_kernel_splitter.cc | 36 StitchInfo GetStitchInfo(const nlohmann::json &kernel_json) { in GetStitchInfo()
52 std::set<std::string> GetRecomputeOps(const nlohmann::json &kernel_json) { in GetRecomputeOps()
130 bool SplitNodesDecoder::DecodeSplitNodes(const nlohmann::json &kernel_json, in DecodeSplitNodes()
760 auto kernel_json = nlohmann::json::parse(json_desc); in DecodeJson() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/kash/ |
| D | kernel_pack.cc | 80 std::ifstream kernel_json(json_f); in ReadFromJsonFile() local
205 std::ifstream kernel_json(json_f); in LoadKernelMeta() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_build.cc | 47 void AkgAscendKernelBuilder::AkgSaveJsonInfo(const string &kernel_name, const string &kernel_json) { in AkgSaveJsonInfo()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_build.cc | 49 void AkgGpuKernelBuilder::AkgSaveJsonInfo(const string &kernel_name, const string &kernel_json) { in AkgSaveJsonInfo()
|
| /third_party/mindspore/tests/ut/cpp/tbe/ |
| D | tbe_json_creator_test.cc | 77 nlohmann::json kernel_json; in TEST_F() local
120 nlohmann::json kernel_json; in TEST_F() local
178 nlohmann::json kernel_json; in TEST_F() local
232 nlohmann::json kernel_json; in TEST_F() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 62 nlohmann::json kernel_json; in TbeOpParallelBuild() local
299 int ParallelBuildManager::StartCompileOp(const nlohmann::json &kernel_json) { in StartCompileOp()
304 std::string ParallelBuildManager::ProcessTbeJob(const nlohmann::json &kernel_json) { in ProcessTbeJob()
|
| D | ascend_kernel_compile.cc | 406 auto kernel_json = iter->second; in QueryFinishJob() local
436 auto kernel_json = iter->second; in QueryFusionFinishJob() local
551 nlohmann::json kernel_json; in AscendPreBuild() local
598 nlohmann::json kernel_json; in AscendSingleOpCompile() local
|
| D | tbe_utils.cc | 387 nlohmann::json kernel_json; in GetCompileInfo() local
|
| D | tbe_kernel_build.cc | 189 nlohmann::json *kernel_json) { in GenTbeSingleKernelJson()
879 bool TbeKernelBuild::GetIOSize(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_l… in GetIOSize()
974 bool TbeKernelBuild::GetIOSize2(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_… in GetIOSize2()
|
| /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()
|
| D | tbe_json_creator.cc | 242 void TbeJsonCreator::GenFusionOpName(nlohmann::json *kernel_json, std::string prefix) { in GenFusionOpName()
286 void TbeJsonCreator::AddOpNameForComputeNode(nlohmann::json *kernel_json) { in AddOpNameForComputeNode()
|
| D | single_tbe_json_creator.cc | 31 bool SingleTbeJsonCreator::GenJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) { in GenJson()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | common_utils.cc | 186 bool KernelMeta::Insert(const std::string &kernel_name, const std::string &kernel_json) { in Insert()
201 std::string kernel_json = bin_map->Search(kernel_name); in CheckCache() local
219 std::string kernel_json = bin_map->Search(kernel_name); in SearchCache() local
238 std::string kernel_json; in InsertCache() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.cc | 529 bool AkgKernelJsonGenerator::CollectJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json) { in CollectJson()
565 nlohmann::json *kernel_json) { in GenStitchJson()
590 … const std::vector<AnfNodePtr> &output_list, nlohmann::json *kernel_json) { in CollectFusedJson()
736 nlohmann::json *kernel_json) { in GenParallelJson()
|
| D | akg_kernel_json_decoder.cc | 388 FuncGraphPtr AkgKernelJsonDecoder::DecodeFusedNodes(const nlohmann::json &kernel_json) { in DecodeFusedNodes()
431 auto kernel_json = nlohmann::json::parse(kernel_json_str); in DecodeFusedNodes() local
|
| D | akg_kernel_json_generator.h | 103 nlohmann::json kernel_json() const { return kernel_json_; } in kernel_json() function
|
| D | akg_kernel_build.cc | 449 auto kernel_json = json_generator.kernel_json_str(); in GetKernelJsonsByHashId() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/ |
| D | tbe_kernel_select.cc | 275 nlohmann::json kernel_json; in TbeCheckSupported() local
449 nlohmann::json kernel_json; in OpSelectFormat() local
|