diff --git a/tensorflow/python/eager/python_eager_op_gen.cc b/tensorflow/python/eager/python_eager_op_gen.cc
index 493f549b35548df074eeacb7845c971ba819b304..511ce82eeba54c574b49b7992a7e6e2f41feeb1b 100644
--- a/tensorflow/python/eager/python_eager_op_gen.cc
+++ b/tensorflow/python/eager/python_eager_op_gen.cc
@@ -277,9 +277,7 @@ string GenEagerPythonOp::Code() {
   AddDocStringArgs();
   AddDocStringInputs();
   AddDocStringAttrs();
-  strings::StrAppend(
-      &result_,
-      "    name: A name for the operation (optional, only for graph mode).\n");
+  AddDocStringNameArg();
   AddOutputGlobals();
   AddDocStringOutputs();
   strings::StrAppend(&result_, "  \"\"\"\n");