fix wrong usage of mshadow::Shape1" (apache#43)

reminisce · May 19, 2017 · 5be8d94 · 5be8d94
1 parent 3ad88f8
commit 5be8d94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/tensor/init_op.h b/src/operator/tensor/init_op.h
@@ -144,7 +144,7 @@ template<typename xpu>
 void FillZerosCsrImpl(mshadow::Stream<xpu> *s, NDArray *dst) {
   if (!dst->storage_initialized()) return;
   // reset the shapes if it's not zeros
-  mshadow::Shape1 new_shape(0);
+  TShape new_shape(mshadow::Shape1(0));
   dst->SetAuxShape(csr::kIndPtr, new_shape);
   dst->SetAuxShape(csr::kIdx, new_shape);
   dst->SetStorageShape(new_shape);