From 55991822a09ac1d675899567ff867cd6feff79c4 Mon Sep 17 00:00:00 2001
From: xzl <zlx_hg@163.com>
Date: Fri, 8 Sep 2017 22:51:14 +0800
Subject: [PATCH] modify GetAttr to Attr

---
 paddle/operators/transpose_op.cc | 2 +-
 paddle/operators/transpose_op.cu | 4 ++--
 paddle/operators/transpose_op.h  | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/paddle/operators/transpose_op.cc b/paddle/operators/transpose_op.cc
index 9b7812c79d3..ea6b2a9ec5a 100644
--- a/paddle/operators/transpose_op.cc
+++ b/paddle/operators/transpose_op.cc
@@ -28,7 +28,7 @@ class TransposeOp : public framework::OperatorWithKernel {
  protected:
   void InferShape(const framework::InferShapeContext &ctx) const override {
     auto in_dim = ctx.Input<Tensor>("X")->dims();
-    auto axis = ctx.GetAttr<std::vector<int>>("axis");
+    auto axis = ctx.Attr<std::vector<int>>("axis");
     size_t in_dim_size = in_dim.size();
     size_t axis_size = axis.size();
 
diff --git a/paddle/operators/transpose_op.cu b/paddle/operators/transpose_op.cu
index 853659e3c3e..24feeea4b2c 100644
--- a/paddle/operators/transpose_op.cu
+++ b/paddle/operators/transpose_op.cu
@@ -98,7 +98,7 @@ class TransposeCUDAKernel : public framework::OpKernel {
                    "It must use GPUPlace.");
     auto* in = context.Input<framework::Tensor>("X");
     auto* out = context.Output<framework::Tensor>("Out");
-    auto axis = context.GetAttr<std::vector<int>>("axis");
+    auto axis = context.Attr<std::vector<int>>("axis");
     TransposeCUDA<T>(context, *in, *out, axis);
   }
 };
@@ -111,7 +111,7 @@ class TransposeGradCUDAKernel : public framework::OpKernel {
                    "It must use GPUPlace.");
     auto* in = context.Input<framework::Tensor>(framework::GradVarName("Out"));
     auto* out = context.Output<framework::Tensor>(framework::GradVarName("X"));
-    auto axis_temp = context.GetAttr<std::vector<int>>("axis");
+    auto axis_temp = context.Attr<std::vector<int>>("axis");
 
     std::vector<int> axis(axis_temp);
 
diff --git a/paddle/operators/transpose_op.h b/paddle/operators/transpose_op.h
index ca64b5a6364..57f63e60e9a 100644
--- a/paddle/operators/transpose_op.h
+++ b/paddle/operators/transpose_op.h
@@ -77,7 +77,7 @@ class TransposeKernel : public framework::OpKernel {
     auto* out = context.Output<framework::Tensor>("Out");
     out->mutable_data<T>(context.GetPlace());
 
-    auto axis = context.GetAttr<std::vector<int>>("axis");
+    auto axis = context.Attr<std::vector<int>>("axis");
     int ndims = axis.size();
     switch (ndims) {
       case 2:
@@ -107,7 +107,7 @@ class TransposeGradKernel : public framework::OpKernel {
     auto* out = context.Output<framework::Tensor>(framework::GradVarName("X"));
     out->mutable_data<T>(context.GetPlace());
 
-    auto axis_temp = context.GetAttr<std::vector<int>>("axis");
+    auto axis_temp = context.Attr<std::vector<int>>("axis");
     std::vector<int> axis(axis_temp);
 
     for (size_t i = 0; i < axis.size(); i++) {
-- 
GitLab