Home
last modified time | relevance | path

Searched refs:kBatchNormGradWithActivation (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.h53 {kBatchNormGradWithActivation, {{0, 1, 7}, {0}}},
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_relu_grad_fusion.cc91 auto prim = std::make_shared<Primitive>(kBatchNormGradWithActivation); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dbatch_norm_grad_gpu_kernel.h122 } else if (kernel_name == kBatchNormGradWithActivation) { in Init()
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h59 constexpr auto kBatchNormGradWithActivation = "BatchNormGradWithActivation"; variable