diff --git a/nnvm/include/nnvm/graph.h b/nnvm/include/nnvm/graph.h index 3f8a2a3642b1..dc8aea221d35 100644 --- a/nnvm/include/nnvm/graph.h +++ b/nnvm/include/nnvm/graph.h @@ -92,7 +92,7 @@ class IndexedGraph { /*! \brief the source node id in the computation graph */ uint32_t node_id; /*! \brief index of output from the source. */ - uint32_t index; + int64_t index; /*! \brief version of the node */ uint32_t version; };