Browse Source

modified MaximumGrad

tags/v1.2.0-rc1
zhengqihao 4 years ago
parent
commit
f3a1a69ce5
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/maximum_grad_cpu_kernel.cc

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

@@ -63,7 +63,7 @@ void MaximumGradRecTask(T *x, T *y, T *dout, T *dx, T *dy, size_t dim, size_t x_
size_t dout_i = i * dout_cargo[dim]; size_t dout_i = i * dout_cargo[dim];


if (dim == dout_shape.size() - 1) { if (dim == dout_shape.size() - 1) {
if (*(x + x_index + x_i) >= *(y + y_index + y_i)) {
if (*(x + x_index + x_i) > *(y + y_index + y_i)) {
*(dx + x_index + x_i) += *(dout + dout_index + i); *(dx + x_index + x_i) += *(dout + dout_index + i);
} else { } else {
*(dy + y_index + y_i) += *(dout + dout_index + i); *(dy + y_index + y_i) += *(dout + dout_index + i);


Loading…
Cancel
Save