From 561234949dc6173320a9affb043020a68ba814a9 Mon Sep 17 00:00:00 2001 From: Sukriti Ramesh Date: Wed, 21 Sep 2016 10:19:40 -0800 Subject: Add AssetFile proto to meta graph def. Change: 133852423 --- tensorflow/core/protobuf/meta_graph.proto | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'tensorflow/core/protobuf/meta_graph.proto') 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 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; +} -- cgit v1.2.3