code refactoring

apache · Aug 2, 2022 · 18a42e5 · 18a42e5
1 parent 5437747
commit 18a42e5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/operator/numpy/np_repeat_op-inl.h b/src/operator/numpy/np_repeat_op-inl.h
@@ -268,8 +268,7 @@ void NumpyRepeatsOpForward(const nnvm::NodeAttrs& attrs,
  void* swap_output_tmp_dptr = temp_space.dptr_;
  void* repeat_output_tmp_dptr = temp_space.dptr_ + inputs[0].shape_.Size() * type_size;
  int* repeat_tmp_dptr =
- reinterpret_cast<int*>(temp_space.dptr_ + inputs[0].shape_.Size() * type_size +
- outputs[0].shape_.Size() * type_size);
+ reinterpret_cast<int*>(repeat_output_tmp_dptr + outputs[0].shape_.Size() * type_size);
 
  // Specify parameters for swapaxis function
  SwapAxisParam swap_axis_param{};