Home
last modified time | relevance | path

Searched defs:kernel_name (Results 1 – 25 of 159) sorted by relevance

1234567

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/
Dgpu_kernel_factory.cc34 void GpuKernelFactory::Register(const std::string &kernel_name, const KernelAttr &kernel_attr, in Register()
39 bool GpuKernelFactory::CheckIOParam(const std::string &kernel_name, const KernelBuildInfo *kernel_i… in CheckIOParam()
55 std::string GpuKernelFactory::SupportedTypeList(const std::string &kernel_name) { in SupportedTypeList()
79 …const std::string &kernel_name, std::shared_ptr<mindspore::kernel::KernelBuildInfo::KernelBuildInf… in ReducePrecision()
126 std::pair<bool, size_t> GpuKernelFactory::GpuKernelAttrCheck(const std::string &kernel_name, in GpuKernelAttrCheck()
180 GpuKernel *GpuKernelFactory::Create(const std::string &kernel_name, const CNodePtr &apply_kernel) { in Create()
192 bool GpuKernelFactory::SearchRegistered(const std::string &kernel_name, const KernelBuildInfoPtr &k… in SearchRegistered()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ascend/
Dakg_ascend_kernel_build.cc29 KernelPackPtr AkgAscendKernelBuilder::AkgSearchCache(const std::string &kernel_name) { in AkgSearchCache()
33 KernelPackPtr AkgAscendKernelBuilder::AkgInsertCache(const std::string &kernel_name) { in AkgInsertCache()
47 void AkgAscendKernelBuilder::AkgSaveJsonInfo(const string &kernel_name, const string &kernel_json) { in AkgSaveJsonInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/gpu/
Dakg_gpu_kernel_build.cc31 KernelPackPtr AkgGpuKernelBuilder::AkgSearchCache(const std::string &kernel_name) { in AkgSearchCache()
35 KernelPackPtr AkgGpuKernelBuilder::AkgInsertCache(const std::string &kernel_name) { in AkgInsertCache()
49 void AkgGpuKernelBuilder::AkgSaveJsonInfo(const string &kernel_name, const string &kernel_json) { in AkgSaveJsonInfo()
/third_party/mindspore/mindspore/ops/_op_impl/_custom_op/
Dcorrection_mul_grad.py52 def correction_mul_grad_compute(dout, x, batch_std, running_std, channel, data_format, kernel_name=… argument
65 def correction_mul_grad(dout, x, batch_std, running_std, dx, mul_dx, channel, kernel_name="correcti… argument
144 def correction_mul_grad_reduce_compute(mul_dx, channel, data_format, kernel_name="correction_mul"): argument
158 def correction_mul_grad_reduce(mul_dx, d_batch_std, channel, kernel_name="correction_mul_grad_reduc… argument
Dfake_learned_scale_quant_perlayer.py52 kernel_name="fake_learned_scale_quant_perlayer"): argument
75 kernel_name="fake_learned_scale_quant_perlayer"): argument
112 kernel_name="fake_learned_scale_quant_perlayer"): argument
Dfake_learned_scale_quant_perchannel.py52 kernel_name="fake_learned_scale_quant_perchannel"): argument
75 kernel_name="fake_learned_scale_quant_perchannel"): argument
120 kernel_name="fake_learned_scale_quant_perchannel"): argument
Dfake_quant_perchannel.py52 kernel_name="fake_quant_perchannel"): argument
90 kernel_name="fake_quant_perchannel"): argument
131 kernel_name="fake_quant_perchannel"): argument
Dfake_learned_scale_quant_perlayer_grad_reduce.py51 … kernel_name="fake_learned_scale_quant_perlayer_" argument
61 … kernel_name="fake_learned_scale_quant_perlayer_grad_d_reduce"): argument
Dfake_learned_scale_quant_perchannel_grad_reduce.py50 … kernel_name="fake_learned_scale_quant_perchannel_" argument
62 … kernel_name="fake_learned_scale_quant_perchannel_grad_d_reduce"): argument
Dfake_learned_scale_quant_perlayer_grad.py94 … kernel_name="fake_learned_scale_quant_perlayer_grad_d"): argument
145 … kernel_name="fake_learned_scale_quant_perlayer_grad_d"): argument
183 … kernel_name="fake_learned_scale_quant_perlayer_grad_d"): argument
Dfake_quant_perchannel_grad.py81 kernel_name="fake_quant_perchannel_grad"): argument
114 kernel_name="fake_quant_perchannel_grad"): argument
155 kernel_name="fake_quant_perchannel_grad"): argument
Dmatmul_cube_dense_left_impl.py50 def shape_gen1(input_x1, input_x2, output_y, kernel_name, trans_a, trans_b): argument
150 def core(shape_a_temp, shape_b_temp, shape_output, kernel_name): argument
215 kernel_name="cus_matmul_cube_dense_left"): argument
Dfake_learned_scale_quant_perchannel_grad.py93 … kernel_name="fake_learned_scale_quant_perchannel_grad_d"): argument
144 … kernel_name="fake_learned_scale_quant_perchannel_grad_d"): argument
190 … channel_axis, kernel_name="fake_learned_scale_quant_perchannel_grad_d"): argument
Dbatchnorm_fold2_grad_reduce.py49 def batchnorm_fold2_grad_reduce_compute(dout, x, dout_args, kernel_name="batchnorm_fold2_grad_reduc… argument
77 def batchnorm_fold2_grad_reduce(dout, x, dout_reduce, dout_x_reduce, kernel_name="batchnorm_fold2_g… argument
Dcorrection_mul.py52 def correction_mul_compute(x, batch_std, running_std, kernel_name="correction_mul"): argument
62 def correction_mul(x, batch_std, running_std, y, channel, kernel_name="correction_mul"): argument
Dbatchnorm_fold2.py58 def batchnorm_fold2_compute(x, beta, gamma, batch_std, batch_mean, running_std, kernel_name="batchn… argument
73 def batchnorm_fold2(x, beta, gamma, batch_std, batch_mean, running_std, y, kernel_name="batchnorm_f… argument
Dfake_quant_perlayer.py54 kernel_name="fake_quant_per_layer"): argument
92 kernel_name="fake_quant_per_layer"): argument
Dbatchnorm_fold2_grad.py58 kernel_name="batchnorm_fold2_grad"): argument
85 d_batch_mean, d_gamma, dx, kernel_name="batchnorm_fold2_grad"): argument
Dfake_quant_perlayer_grad.py82 kernel_name="fake_quant_per_layer_grad"): argument
118 kernel_name="fake_quant_per_layer_grad"): argument
/third_party/mindspore/mindspore/common/
Dseed.py179 def _update_seeds(op_seed, kernel_name): argument
192 def _get_op_seed(op_seed, kernel_name): argument
222 def _get_graph_seed(op_seed, kernel_name): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc37 void CPUKernelFactory::Register(const std::string &kernel_name, const KernelAttr &kernel_attr, in Register()
45 std::shared_ptr<CPUKernel> CPUKernelFactory::Create(const std::string &kernel_name, const CNodePtr … in Create()
90 void CPUKernelFactory::UpdateKernelAttrs(const std::string &kernel_name, const std::vector<KernelAt… in UpdateKernelAttrs()
114 std::pair<bool, size_t> CPUKernelFactory::CPUKernelAttrCheck(const std::string &kernel_name, in CPUKernelAttrCheck()
165 std::vector<KernelAttr> CPUKernelFactory::GetSupportedKernelAttrList(const std::string &kernel_name… in GetSupportedKernelAttrList()
/third_party/mindspore/mindspore/ccsrc/debug/data_dump/
Dcpu_e2e_dump.cc26 std::string kernel_name = GetKernelNodeName(node); in DumpCNodeData() local
44 std::string kernel_name = GetKernelNodeName(node); in DumpCNodeInputs() local
52 std::string kernel_name = GetKernelNodeName(node); in DumpCNodeOutputs() local
58 …Dump::DumpInputImpl(const CNodePtr &node, const std::string &dump_path, std::string *kernel_name) { in DumpInputImpl()
85 …ump::DumpOutputImpl(const CNodePtr &node, const std::string &dump_path, std::string *kernel_name) { in DumpOutputImpl()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dcommon_utils.h148 #define CHECK_KERNEL_INPUTS_NUM(actual_inputs_num, expect_inputs_num, kernel_name) … argument
156 #define CHECK_KERNEL_OUTPUTS_NUM(actual_outputs_num, expect_outputs_num, kernel_name) … argument
164 #define CHECK_KERNEL_WORKSPACE_SIZE(actual_size, expect_size, kernel_name) … argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_build.cc259 auto kernel_name = json_generator.kernel_name(); in AddKernels() local
388 auto kernel_name = json_generator.kernel_name(); in GetNotCachedKernels() local
410 auto kernel_name = json_generator.kernel_name(); in InsertToCache() local
425 auto kernel_name = json_generator.kernel_name(); in HandleRepeatNodes() local
444 auto kernel_name = json_generator.kernel_name(); in GetKernelJsonsByHashId() local
/third_party/mesa3d/src/gallium/frontends/clover/llvm/codegen/
Dcommon.cpp109 const std::string &kernel_name) { in get_reqd_work_group_size()
118 kernel_attributes(const Module &mod, const std::string &kernel_name) { in kernel_attributes()
151 make_kernel_args(const Module &mod, const std::string &kernel_name, in make_kernel_args()

1234567