diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.cc index cb9b253de2..8121d6f337 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.cc @@ -59,11 +59,10 @@ void DropoutGradCpuBwdKernel::DropoutBackwardKernel(const std::vector(inputs[0]->addr); auto mask = reinterpret_cast(inputs[1]->addr); - float scale = 1.f / keep_prob; + const float scale = 1.f / keep_prob; for (size_t i = 0; i < num_count; i += 1) { dx[i] = (T)(scale * static_cast(dy[i] * mask[i])); } } - } // namespace kernel } // namespace mindspore diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.h index af55615335..a7a3fc9ff3 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.h @@ -26,7 +26,6 @@ namespace mindspore { namespace kernel { - class DropoutGradCpuBwdKernel : public CPUKernel { public: DropoutGradCpuBwdKernel() = default; @@ -36,9 +35,9 @@ class DropoutGradCpuBwdKernel : public CPUKernel { const std::vector &outputs) override; private: - float keep_prob_{0.0}; + float keep_prob_{1.0}; size_t num_count_{1}; - TypeId dtype_; + TypeId dtype_{kTypeUnknown}; template void DropoutBackwardKernel(const std::vector &inputs, const std::vector &outputs, size_t num_count, float keep_prob);