!10015 static check warning clean for dropout grad

From: @xutianming1985
Reviewed-by: @c_34,@wuxuejian
Signed-off-by: @c_34
pull/10015/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit 3f9e90a0a8

@ -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

@ -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