index tensor to have int64

apache · Sep 11, 2019 · 5b6d697 · 5b6d697
1 parent f8f4300
commit 5b6d697
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/operator/sequence_last.cc b/src/operator/sequence_last.cc
@@ -31,9 +31,9 @@ template <>
 Operator *CreateOp<cpu>(SequenceLastParam param, int dtype, int itype) {
   Operator *op = nullptr;
   MSHADOW_TYPE_SWITCH(dtype, DType, {
-      MSHADOW_TYPE_SWITCH(itype, IType, {
-          op = new SequenceLastOp<cpu, DType, IType>(param);
-        });
+//      MSHADOW_TYPE_SWITCH(itype, IType, {
+          op = new SequenceLastOp<cpu, DType, int64_t>(param);
+  //      });
     });
   return op;
 }