Home
last modified time | relevance | path

Searched defs:kernel_json (Results 1 – 14 of 14) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/akg/
Dakg_cpu_kernel_build.cc32 void AkgCpuKernelBuilder::SaveJsonInfo(const string &kernel_name, const string &kernel_json) { in SaveJsonInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/akg/
Dakg_gpu_kernel_build.cc42 void AkgGpuKernelBuilder::SaveJsonInfo(const string &kernel_name, const string &kernel_json) { in SaveJsonInfo()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/tbe/
Dtbe_json_creator_test.cc81 nlohmann::json kernel_json; in TEST_F() local
123 nlohmann::json kernel_json; in TEST_F() local
181 nlohmann::json kernel_json; in TEST_F() local
234 nlohmann::json kernel_json; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/akg/
Dakg_ascend_kernel_build.cc51 void AkgAscendKernelBuilder::SaveJsonInfo(const string &kernel_name, const string &kernel_json) { in SaveJsonInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/dynamic_akg/
Ddynamic_akg_gpu_kernel_build.cc56 void DynamicAkgGpuKernelBuilder::SaveJsonInfo(const string &kernel_name, const string &kernel_json)… in SaveJsonInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/dynamic_akg/
Ddynamic_akg_cpu_kernel_build.cc49 void DynamicAkgCpuKernelBuilder::SaveJsonInfo(const string &kernel_name, const string &kernel_json)… in SaveJsonInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/
Dgraph_kernel_builder.cc476 auto kernel_json = json_generator.kernel_json_str(); in GetKernelJsonsByHashId() local
498 std::string kernel_json = entry->d_name; in LoadCache() local
Dgraph_kernel_json_generator.h81 nlohmann::json kernel_json() const { return kernel_json_; } in kernel_json() function
Dgraph_kernel_json_generator.cc393 void GraphKernelJsonGenerator::SaveShape(const AnfNodePtr &node, nlohmann::json *kernel_json, in SaveShape()
851 bool GraphKernelJsonGenerator::CollectJson(const AnfNodePtr &anf_node, nlohmann::json *kernel_json)… in CollectJson()
914 …JsonGenerator::GenKernelName(const FuncGraphPtr &fg, size_t hash_id, nlohmann::json *kernel_json) { in GenKernelName()
938 … const std::vector<AnfNodePtr> &output_list, nlohmann::json *kernel_json, in CollectFusedJson()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dgraph_kernel_splitter_with_py.cc134 …bool DecodeSplitNodes(const nlohmann::json &kernel_json, const std::map<std::string, AnfNodePtr> &… in DecodeSplitNodes()
223 auto kernel_json = nlohmann::json::parse(json_desc); in DecodeJson() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/akg/
Dakg_kernel_json_decoder.cc376 FuncGraphPtr AkgKernelJsonDecoder::DecodeFusedNodes(const nlohmann::json &kernel_json) { in DecodeFusedNodes()
426 auto kernel_json = nlohmann::json::parse(kernel_json_str); in DecodeFusedNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc234 std::string kernel_json = bin_map->Search(kernel_name); in CheckCache() local
252 std::string kernel_json = bin_map->Search(kernel_name); in SearchCache() local
275 std::string kernel_json = bin_map->kernel_meta_path(); in InsertCache() local
308 bool KernelMeta::Insert(const std::string &kernel_name, const std::string &kernel_json) { in Insert()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_build.cc260 std::ifstream kernel_json(json_path); in CollectNotCachedNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/kash/
Dkernel_pack.cc162 std::ifstream kernel_json(json_f); in ReadFromJsonFile() local