enable negative values in TShape (#125)

5973da05 · Sheng Zha · Tianqi Chen · 06c06f58 · 5973da05
Commit 5973da05 authored Jul 06, 2017 by Sheng Zha Committed by Tianqi Chen May 29, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

nnvm/include/nnvm/tuple.h
+1 -1

No files found.
--- a/nnvm/include/nnvm/tuple.h
+++ b/nnvm/include/nnvm/tuple.h
@@ -216,7 +216,7 @@ class Tuple {
    // get (
    while (true) {
      char ch = is.peek();
-      if (isdigit(ch)) {
+      if (isdigit(ch) || ch == '-') {
        ValueType idx;
        if (is >> idx) {
          t.assign(&idx, &idx + 1);