Address cr

apache · Nov 6, 2017 · dc17857 · dc17857
1 parent bce0c01
commit dc17857
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/operator/tensor/indexing_op.cc b/src/operator/tensor/indexing_op.cc
@@ -322,7 +322,8 @@ Examples::
 
   data = [2, 3, 0]
   indices = [[1, 1, 0], [0, 1, 0]]
-  scatter_nd(data, indices) = [[0, 0], [2, 3]]
+  shape = (2, 2)
+  scatter_nd(data, indices, shape) = [[0, 0], [2, 3]]
 
 )code")
 .set_num_outputs(1)

diff --git a/src/operator/tensor/matrix_op-inl.h b/src/operator/tensor/matrix_op-inl.h
@@ -675,7 +675,7 @@ struct slice_forward {
         stride *= dshape[k];
       }
       KERNEL_ASSIGN(out[out_offset++], req,
-                    data[irow*data_last_dim_size+j*step_last_dim+begin_last_dim]);
+                    data[irow * data_last_dim_size + j * step_last_dim + begin_last_dim]);
     }
   }
 };
@@ -731,7 +731,7 @@ struct slice_backward {
         idx /= oshape[k];
         stride *= dshape[k];
       }
-      KERNEL_ASSIGN(igrad[irow*data_last_dim_size+j*step_last_dim+begin_last_dim],
+      KERNEL_ASSIGN(igrad[irow * data_last_dim_size + j * step_last_dim + begin_last_dim],
                     req, ograd[ograd_offset++]);
     }
   }