experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix update_api_def.sh script to output ApiDefs instead of ApiDef proto.

Browse Source 
PiperOrigin-RevId: 183109303
This commit is contained in:
Anna R 2018-01-24 10:55:55 -08:00 committed by TensorFlower Gardener
parent 18cb45c1cc
commit 5031e9f169
3 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/api_def/update_api_def.cc
View File

@ -224,14 +224,14 @@ void RemoveDocs(const std::vector<const OpDef*>& ops,
}
} // namespace
// Returns ApiDef text representation in multi-line format
// Returns ApiDefs text representation in multi-line format
// constructed based on the given op.
string CreateApiDef(const OpDef& op) {
ApiDef api_def;
FillBaseApiDef(&api_def, op);
ApiDefs api_defs;
FillBaseApiDef(api_defs.add_op(), op);
const std::vector<string> multi_line_fields = {"description"};
string new_api_defs_str = api_def.DebugString();
string new_api_defs_str = api_defs.DebugString();
return PBTxtToMultiline(new_api_defs_str, multi_line_fields);
}

2
tensorflow/core/api_def/update_api_def.h
View File

@ -21,7 +21,7 @@ limitations under the License.
namespace tensorflow {
// Returns ApiDef text representation in multi-line format
// Returns ApiDefs text representation in multi-line format
// constructed based on the given op.
string CreateApiDef(const OpDef& op);

32
tensorflow/core/api_def/update_api_def_test.cc
View File

@ -173,30 +173,32 @@ description: "Description\nfor Op1."
OpDef op;
protobuf::TextFormat::ParseFromString(op_text, &op); // NOLINT
const string expected_api_def = R"(graph_op_name: "Op1"
in_arg {
name: "a"
description: <<END
const string expected_api_def = R"(op {
graph_op_name: "Op1"
in_arg {
name: "a"
description: <<END
Description for a.
END
}
out_arg {
name: "output"
description: <<END
}
out_arg {
name: "output"
description: <<END
Description for output.
END
}
attr {
name: "b"
description: <<END
}
attr {
name: "b"
description: <<END
Description for b.
END
}
summary: "Summary for Op1."
description: <<END
}
summary: "Summary for Op1."
description: <<END
Description
for Op1.
END
}
)";
EXPECT_EQ(expected_api_def, CreateApiDef(op));
}