Home
last modified time | relevance | path

Searched refs:kBatchNormWithAddAndActivation (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.h51 {kBatchNormWithAddAndActivation, {{0, 5}, {0}}},
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_add_relu_fusion.cc79 auto prim = std::make_shared<Primitive>(kBatchNormWithAddAndActivation); in Process()
Dpost_batch_norm_add_relu_fusion.cc79 auto prim = std::make_shared<Primitive>(kBatchNormWithAddAndActivation); in Process()
Dbatch_norm_add_relu_grad_fusion.cc143 if (AnfAlgo::GetCNodeName(forward_node) != kBatchNormWithAddAndActivation) { in PatternCheck()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dbatch_norm_gpu_kernel.h92 } else if (kernel_name == kBatchNormWithAddAndActivation) { in Init()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h58 constexpr auto kBatchNormWithAddAndActivation = "BatchNormWithAddAndActivation"; variable