Fixed rnn_memory_helper_op.cc

b5a3d1e5 · Kavya Srinet · ec1d63af · b5a3d1e5
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

paddle/fluid/operators/rnn_memory_helper_op.cc paddle/fluid/operators/rnn_memory_helper_op.cc +2 -2

未找到文件。
--- a/paddle/fluid/operators/rnn_memory_helper_op.cc
+++ b/paddle/fluid/operators/rnn_memory_helper_op.cc
@@ -66,7 +66,7 @@ class RNNMemoryHelperOpInfoMaker : public framework::OpProtoAndCheckerMaker {
    AddAttr<int>("dtype",
                 "(int, default 5 (FP32)) "
                 "Output data type")
-        .SetDefault(framework::proto::DataType::FP32);
+        .SetDefault(framework::proto::VarType::Type::FP32);
    AddComment("");
  }
 };
@@ -126,7 +126,7 @@ class RNNMemoryHelperGradOpInfoMaker
    AddAttr<int>("dtype",
                 "(int, default 5 (FP32)) "
                 "Output data type")
-        .SetDefault(framework::proto::DataType::FP32);
+        .SetDefault(framework::proto::VarType::Type::FP32);
    AddComment("");
  }
 };