| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | relu_v2_pass.cc | 42 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
90 auto build_info = GenerateKernelBuildInfo(new_node); in CreateReluV2()
118 auto build_info = GenerateKernelBuildInfo(new_node); in CreateReluGradV2()
|
| D | add_relu_v2_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
100 auto build_info = GenerateKernelBuildInfo(add_relu); in Process()
|
| D | matmul_biasadd_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
98 auto build_info = GenerateKernelBuildInfo(fused_node); in Process()
|
| D | add_relu_grad_v2_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
95 auto build_info = GenerateKernelBuildInfo(add_relugrad); in Process()
|
| D | adam_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
170 auto build_info = GenerateKernelBuildInfo(adam); in Process()
|
| D | print_reduce_fusion.cc | 29 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
187 auto build_info = GenerateKernelBuildInfo(print_fused); in Run()
|
| D | adam_weight_decay_fusion.cc | 30 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(CNodePtr node) { in GenerateKernelBuildInfo() function
176 auto build_info = GenerateKernelBuildInfo(adam_weight_decay); in Process()
|
| D | concat_outputs_for_all_gather.cc | 50 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const AnfNodePtr &concat, const OutputInfo &allg… in GenerateKernelBuildInfo() function
111 auto kernel_build_info = GenerateKernelBuildInfo(concat, output_info, inputs_size, i); in InsertConcatForOutput()
|
| D | combine_momentum_fusion.cc | 94 auto kernel_build_info = GenerateKernelBuildInfo(momentums); in Run()
|
| D | combine_cast_fusion.cc | 111 auto kernel_build_info = GenerateKernelBuildInfo(cast_list); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | replace_node_by_proxy.cc | 25 kernel::KernelBuildInfoPtr ReplaceNodeByProxy::GenerateKernelBuildInfo(const CNodePtr &cnode) { in GenerateKernelBuildInfo() function in mindspore::opt::ReplaceNodeByProxy
84 auto kernel_build_info = GenerateKernelBuildInfo(cnode); in Run()
|
| D | replace_node_by_proxy.h | 37 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const CNodePtr &cnode);
|
| D | communication_op_fusion.cc | 37 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const CommunicationOpInfo &communication_op_info… in GenerateKernelBuildInfo() function
369 auto kernel_build_info = GenerateKernelBuildInfo(communication_op_info, start_index, end_index); in CreateFusedCommunicationOp()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | util.h | 62 …static kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const std::vector<AnfNodePtr> &node_list…
|
| D | util.cc | 182 auto kernel_build_info = GenerateKernelBuildInfo(single_nodes); in DoFusion()
205 kernel::KernelBuildInfoPtr Util::GenerateKernelBuildInfo(const std::vector<AnfNodePtr> &node_list) { in GenerateKernelBuildInfo() function in mindspore::ps::Util
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | concat_outputs_for_all_gather.cc | 50 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const AnfNodePtr &concat, const OutputInfo &allg… in GenerateKernelBuildInfo() function
113 auto kernel_build_info = GenerateKernelBuildInfo(concat, output_info, inputs_size, i); in InsertConcatForOutput()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | helper.h | 225 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const std::vector<AnfNodePtr> &node_list);
|
| D | helper.cc | 964 kernel::KernelBuildInfoPtr GenerateKernelBuildInfo(const std::vector<AnfNodePtr> &node_list) { in GenerateKernelBuildInfo() function
|