Searched refs:AkgKernelJsonGenerator (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.cc | 133 int AkgKernelJsonGenerator::op_cnt_ = 0;
134 std::mutex AkgKernelJsonGenerator::op_cnt_mtx_;
136 int AkgKernelJsonGenerator::GetOpCntInc() { in GetOpCntInc()
143 TypeId AkgKernelJsonGenerator::GetInputDataType(const AnfNodePtr &anf_node, size_t real_index) cons… in GetInputDataType()
148 std::vector<size_t> AkgKernelJsonGenerator::GetInputShape(const AnfNodePtr &anf_node, size_t real_i… in GetInputShape()
153 std::string AkgKernelJsonGenerator::GetInputFormat(const AnfNodePtr &anf_node, size_t real_index) c… in GetInputFormat()
157 TypeId AkgKernelJsonGenerator::GetOutputDataType(const AnfNodePtr &anf_node, size_t index) const { in GetOutputDataType()
162 std::vector<size_t> AkgKernelJsonGenerator::GetOutputShape(const AnfNodePtr &anf_node, size_t index… in GetOutputShape()
167 std::string AkgKernelJsonGenerator::GetOutputFormat(const AnfNodePtr &anf_node, size_t index) const… in GetOutputFormat()
171 bool AkgKernelJsonGenerator::CreateInputDescJson(const AnfNodePtr &anf_node, const OpInfoPtr &op_in… in CreateInputDescJson()
[all …]
|
| D | akg_kernel_json_generator.h | 89 class AkgKernelJsonGenerator {
91 AkgKernelJsonGenerator() { Clear(); } in AkgKernelJsonGenerator() function
92 explicit AkgKernelJsonGenerator(DumpOption dump_option) : dump_option_(dump_option) { Clear(); } in AkgKernelJsonGenerator() function
93 ~AkgKernelJsonGenerator() = default;
|
| D | akg_kernel_build.h | 34 using JsonNodePair = std::pair<AkgKernelJsonGenerator, AnfNodePtr>;
44 …virtual void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_…
|
| D | akg_kernel_build.cc | 532 AkgKernelJsonGenerator akg_kernel_json_generator(option); in AkgKernelParallelBuild()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/ |
| D | akg_gpu_kernel_build.h | 33 …void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_generato…
|
| D | akg_gpu_kernel_build.cc | 40 … const AkgKernelJsonGenerator &json_generator, const AnfNodePtr &anf_node) { in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/ |
| D | akg_ascend_kernel_build.h | 37 …void AkgSetKernelMod(const KernelPackPtr &kernel_pack, const AkgKernelJsonGenerator &json_generato…
|
| D | akg_ascend_kernel_build.cc | 38 … const AkgKernelJsonGenerator &json_generator, const AnfNodePtr &anf_node) { in AkgSetKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 109 kernel::AkgKernelJsonGenerator json_generator(dump_option); in ExpandJsonInfo()
|
| D | graph_kernel_helper.cc | 103 kernel::AkgKernelJsonGenerator akg_kernel_json_generator(dump_option); in GenJson()
|