Searched refs:kBatchNormGradWithAddAndActivation (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | batch_norm_add_relu_grad_fusion.cc | 89 …MS_LOG(EXCEPTION) << "The output size of node " << kBatchNormGradWithAddAndActivation << " must be… in ReplaceOutput()
192 auto prim = std::make_shared<Primitive>(kBatchNormGradWithAddAndActivation); in Process()
|
| D | cudnn_inplace_fusion.cc | 45 …32_t> kInplaceOpNames = {{kConv2DBackpropInputOpName, 0}, {kBatchNormGradWithAddAndActivation, 3}};
53 …{kConv2DBackpropInputOpName, 0}, {kmaxPoolGradOpName, 2}, {kBatchNormGradWithAddAndActivation, 0}};
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | kernel_info_setter.h | 54 {kBatchNormGradWithAddAndActivation, {{0, 1, 7}, {0, 3}}},
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | batch_norm_grad_gpu_kernel.h | 124 } else if (kernel_name == kBatchNormGradWithAddAndActivation) { in Init()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 60 constexpr auto kBatchNormGradWithAddAndActivation = "BatchNormGradWithAddAndActivation"; variable
|