Fix build failure

Author: wweic

nnvm/include/nnvm/op_attr_types.h

@@ -224,7 +224,7 @@ using FCorrectLayout = std::function<bool(
  * \return success flag.
  */
 using FCorrectLayoutEx = std::function<bool(
-    const NodeAttrs& attrs,
+    NodeAttrs& attrs,
     std::vector<TShape>* ishapes,
     std::vector<Layout>* ilayouts,
     const std::vector<Layout>* last_ilayouts,

nnvm/src/top/nn/nn.cc

@@ -194,7 +194,7 @@ inline bool BatchNormInferShape(const nnvm::NodeAttrs& attrs,
   return true;
 }
 
-inline bool BatchNormCorrectLayout(const NodeAttrs& attrs,
+inline bool BatchNormCorrectLayout(NodeAttrs& attrs,
                                    std::vector<TShape>* ishapes,
                                    std::vector<Layout> *in_layouts,
                                    const std::vector<Layout> *last_in_layouts,
@@ -593,7 +593,7 @@ inline bool PadInferShape(const nnvm::NodeAttrs& attrs,
   return true;
 }
 
-inline bool PadCorrectLayout(const NodeAttrs& attrs,
+inline bool PadCorrectLayout(NodeAttrs& attrs,
                              std::vector<TShape>* ishapes,
                              std::vector<Layout>* ilayouts,
                              const std::vector<Layout>* last_ilayouts,