From c7e7e90c820d0075d257bb3b8776e5c478abe48f Mon Sep 17 00:00:00 2001 From: Kavya Srinet Date: Wed, 14 Feb 2018 11:48:10 -0800 Subject: [PATCH] Fixed assign_value_op.h --- paddle/fluid/operators/assign_value_op.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/assign_value_op.h b/paddle/fluid/operators/assign_value_op.h index 6a62ca53d4..cdcd7c406a 100644 --- a/paddle/fluid/operators/assign_value_op.h +++ b/paddle/fluid/operators/assign_value_op.h @@ -30,10 +30,10 @@ class AssignValueKernel : public framework::OpKernel { int dtype = ctx.Attr("dtype"); const char* value_name = nullptr; switch (dtype) { - case framework::proto::VarType::Type::INT32: + case framework::proto::VarType::INT32: value_name = "int32_values"; break; - case framework::proto::VarType::Type::FP32: + case framework::proto::VarType::FP32: value_name = "fp32_values"; break; default: -- GitLab