fix cpplint

src/operator/nn/mkldnn/mkldnn_flatten.cc

@@ -31,7 +31,7 @@ namespace mxnet {
 namespace op {
 
 class MKLDNNFlattenFwd : public MKLDNNReshapeFwd {
-public:
+ public:
   explicit MKLDNNFlattenFwd(const OpReqType &req,
                             const NDArray &input,
                             const NDArray &output)
@@ -81,7 +81,7 @@ void MKLDNNFlattenForward(const nnvm::NodeAttrs &attrs,
   fwd.Execute(input, output, ws_ptr);
 }
 
-}
-}
+} // namespace op
+} // namespace mxnet
 
 #endif

src/operator/nn/mkldnn/mkldnn_reshape-inl.h

@@ -27,6 +27,7 @@
 #define MXNET_OPERATOR_NN_MKLDNN_MKLDNN_RESHAPE_INL_H_
 
 #if MXNET_USE_MKLDNN == 1
+#include <vector>
 #include "mkldnn_base-inl.h"
 #include "../../tensor/matrix_op-inl.h"
 
@@ -46,14 +47,14 @@ inline bool SupportMKLDNNReshape(const ReshapeParam &param,
 }
 
 class MKLDNNReshapeFwd {
-protected:
+ protected:
   std::shared_ptr<mkldnn::memory> data_;
   std::shared_ptr<mkldnn::memory> out_;
   std::shared_ptr<mkldnn::memory> temp_;
   std::vector<mkldnn::primitive> prims_;
   bool needInvalidateInput = false;
 
-public:
+ public:
   MKLDNNReshapeFwd(const OpReqType &req,
                    const NDArray &input,
                    const NDArray &output);
@@ -76,8 +77,8 @@ void MKLDNNReshapeForward(const nnvm::NodeAttrs& attrs,
                           const NDArray &input,
                           const OpReqType &req,
                           const NDArray &output);
-}
-}
+} // namespace op
+} // namespace mxnet
 
-#endif
-#endif
+#endif // MXNET_USE_MKLDNN == 1
+#endif // MXNET_OPERATOR_NN_MKLDNN_MKLDNN_RESHAPE_INL_H_