Add AssetFile proto to meta graph def.

Change: 133852423

1 files changed, 14 insertions, 0 deletions

diff --git a/tensorflow/core/protobuf/meta_graph.proto b/tensorflow/core/protobuf/meta_graph.proto
index cd3dbf86a2..65e1e9fb7d 100644
--- a/tensorflow/core/protobuf/meta_graph.proto
+++ b/tensorflow/core/protobuf/meta_graph.proto
@@ -67,6 +67,9 @@ message MetaGraphDef {
   // signature_def: Map from user supplied key for a signature to a single
   // SignatureDef.
   map<string, SignatureDef> signature_def = 5;
+
+  // Asset file def to be used with the defined graph.
+  repeated AssetFileDef asset_file_def = 6;
 }
 
 // CollectionDef should cover most collections.
@@ -266,3 +269,14 @@ message SignatureDef {
   // where a single graph computation may return multiple results.
   string method_name = 3;
 }
+
+// An asset file def for a single file or a set of sharded files with the same
+// name.
+message AssetFileDef {
+  // The tensor to bind the asset filename to.
+  TensorInfo tensor_info = 1;
+  // The filename within an assets directory. Note: does not include the path
+  // prefix, i.e. directories. For an asset at /tmp/path/vocab.txt, the filename
+  // would be "vocab.txt".
+  string filename = 2;
+}