Fix print_operators_doc to support AttrType int_64

7 years ago · e54eaee58a
parent fd5c1c8a69
commit e54eaee58a
1 changed files with 2 additions and 0 deletions
--- a/paddle/pybind/print_operators_doc.cc
+++ b/paddle/pybind/print_operators_doc.cc
@ -64,6 +64,8 @@ std::string AttrType(paddle::framework::proto::AttrType at) {
      return "bool array";
    case paddle::framework::proto::BLOCK:
      return "block id";
+    case paddle::framework::proto::LONG:
+      return "long";
  }
  return "UNKNOWN";  // not possible
 }