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 dy = reinterpret_cast<T *>(inputs[0]->addr);
auto mask = reinterpret_cast<T *>(inputs[1]->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) { for (size_t i = 0; i < num_count; i += 1) {
dx[i] = (T)(scale * static_cast<float>(dy[i] * mask[i])); dx[i] = (T)(scale * static_cast<float>(dy[i] * mask[i]));
} }
} }

} // namespace kernel } // namespace kernel
} // namespace mindspore } // namespace mindspore

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

@@ -26,7 +26,6 @@


namespace mindspore { namespace mindspore {
namespace kernel { namespace kernel {

class DropoutGradCpuBwdKernel : public CPUKernel { class DropoutGradCpuBwdKernel : public CPUKernel {
public: public:
DropoutGradCpuBwdKernel() = default; DropoutGradCpuBwdKernel() = default;
@@ -36,9 +35,9 @@ class DropoutGradCpuBwdKernel : public CPUKernel {
const std::vector<AddressPtr> &outputs) override; const std::vector<AddressPtr> &outputs) override;


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


Loading…
Cancel
Save