Browse Source

!10015 static check warning clean for dropout grad

From: @xutianming1985
Reviewed-by: @c_34,@wuxuejian
Signed-off-by: @c_34
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
3f9e90a0a8
2 changed files with 3 additions and 5 deletions
  1. +1
    -2
      mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.cc
  2. +2
    -3
      mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.h

+ 1
- 2
mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.cc View File

@@ -59,11 +59,10 @@ void DropoutGradCpuBwdKernel::DropoutBackwardKernel(const std::vector<AddressPtr
auto dy = reinterpret_cast<T *>(inputs[0]->addr);
auto mask = reinterpret_cast<T *>(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<float>(dy[i] * mask[i]));
}
}

} // namespace kernel
} // namespace mindspore

+ 2
- 3
mindspore/ccsrc/backend/kernel_compiler/cpu/dropout_grad_kernel.h View File

@@ -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<AddressPtr> &outputs) override;

private:
float keep_prob_{0.0};
float keep_prob_{1.0};
size_t num_count_{1};
TypeId dtype_;
TypeId dtype_{kTypeUnknown};
template <typename T>
void DropoutBackwardKernel(const std::vector<AddressPtr> &inputs, const std::vector<AddressPtr> &outputs,
size_t num_count, float keep_prob);


Loading…
Cancel
Save