diff --git a/nnvm/include/nnvm/layout.h b/nnvm/include/nnvm/layout.h index bec523b348115..918eb9a25dee0 100644 --- a/nnvm/include/nnvm/layout.h +++ b/nnvm/include/nnvm/layout.h @@ -348,7 +348,7 @@ class Layout { return false; } - inline const LayoutDim operator[](size_t i) const { + inline LayoutDim operator[](size_t i) const { return layout_simplified_[i]; }