@@ -194,11 +194,12 @@ inline bool BatchNormInferShape(const nnvm::NodeAttrs& attrs,
194194 return true ;
195195}
196196
197- inline bool BatchNormCorrectLayout (NodeAttrs& attrs ,
197+ inline bool BatchNormCorrectLayout (const NodeAttrs& attrs_const ,
198198 std::vector<TShape>* ishapes,
199199 std::vector<Layout> *in_layouts,
200200 const std::vector<Layout> *last_in_layouts,
201201 std::vector<Layout> *out_layouts) {
202+ NodeAttrs& attrs = const_cast <NodeAttrs&>(attrs_const);
202203 const BatchNormParam& param = nnvm::get<BatchNormParam>(attrs.parsed );
203204 CHECK_EQ (in_layouts->size (), 5U );
204205 CHECK_EQ (last_in_layouts->size (), 5U );
@@ -593,11 +594,12 @@ inline bool PadInferShape(const nnvm::NodeAttrs& attrs,
593594 return true ;
594595}
595596
596- inline bool PadCorrectLayout (NodeAttrs& attrs ,
597+ inline bool PadCorrectLayout (const NodeAttrs& attrs_const ,
597598 std::vector<TShape>* ishapes,
598599 std::vector<Layout>* ilayouts,
599600 const std::vector<Layout>* last_ilayouts,
600601 std::vector<Layout>* olayouts) {
602+ NodeAttrs& attrs = const_cast <NodeAttrs&>(attrs_const);
601603 const PadParam& param = nnvm::get<PadParam>(attrs.parsed );
602604 const auto & last_layout = last_ilayouts->at (0 );
603605 Layout layout = ilayouts->at (0 );
0 commit comments