Update take_grad.h

pzz2011 · Jun 22, 2016 · 1b5afb0 · 1b5afb0
1 parent 0198f5f
commit 1b5afb0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mshadow/extension/take_grad.h b/mshadow/extension/take_grad.h
@@ -37,11 +37,11 @@ template<typename IndexExp,
          typename SrcExp,
          typename DType,
          int e1, int e2>
-inline TakeGradExp<IndexExp, SrcExp, default_real_t>
+inline TakeGradExp<IndexExp, SrcExp, DType>
 take_grad(const Exp<IndexExp, DType, e1> &index,
           const Exp<SrcExp, DType, e2> &src,
           const index_t input_dim) {
-  return TakeGradExp<IndexExp, SrcExp, default_real_t>(index.self(),
+  return TakeGradExp<IndexExp, SrcExp, DType>(index.self(),
                                                        src.self(),
                                                        input_dim);
 }