diff options
author | 2018-09-05 15:24:38 -0700 | |
---|---|---|
committer | 2018-09-05 15:42:44 -0700 | |
commit | 47b1af2a3a724a5d783ae06ca0e0e78b30e0799b (patch) | |
tree | 6a7595d1210fbb18a2d2e6adf5fc3b47caf644bf /tensorflow/tools | |
parent | ebf6d259fd4c57114c17646e40fdcfa4a1472972 (diff) |
Expose an axis argument for VocabInfo, which allows for warm-starting of the second axis of Tensors through tf.train.warm_start. Note that the underlying initializer already has this functionality (for example, for output layers).
PiperOrigin-RevId: 211709879
Diffstat (limited to 'tensorflow/tools')
4 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/tools/api/golden/v1/tensorflow.estimator.-vocab-info.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.estimator.-vocab-info.pbtxt index 5301b94eb3..b6942cb7ed 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.estimator.-vocab-info.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.estimator.-vocab-info.pbtxt @@ -4,6 +4,10 @@ tf_class { is_instance: "<class \'tensorflow.python.training.warm_starting_util.VocabInfo\'>" is_instance: "<type \'tuple\'>" member { + name: "axis" + mtype: "<type \'property\'>" + } + member { name: "backup_initializer" mtype: "<type \'property\'>" } diff --git a/tensorflow/tools/api/golden/v1/tensorflow.train.-vocab-info.pbtxt b/tensorflow/tools/api/golden/v1/tensorflow.train.-vocab-info.pbtxt index 4ce7cb1111..39b946b82f 100644 --- a/tensorflow/tools/api/golden/v1/tensorflow.train.-vocab-info.pbtxt +++ b/tensorflow/tools/api/golden/v1/tensorflow.train.-vocab-info.pbtxt @@ -4,6 +4,10 @@ tf_class { is_instance: "<class \'tensorflow.python.training.warm_starting_util.VocabInfo\'>" is_instance: "<type \'tuple\'>" member { + name: "axis" + mtype: "<type \'property\'>" + } + member { name: "backup_initializer" mtype: "<type \'property\'>" } diff --git a/tensorflow/tools/api/golden/v2/tensorflow.estimator.-vocab-info.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.estimator.-vocab-info.pbtxt index 5301b94eb3..b6942cb7ed 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.estimator.-vocab-info.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.estimator.-vocab-info.pbtxt @@ -4,6 +4,10 @@ tf_class { is_instance: "<class \'tensorflow.python.training.warm_starting_util.VocabInfo\'>" is_instance: "<type \'tuple\'>" member { + name: "axis" + mtype: "<type \'property\'>" + } + member { name: "backup_initializer" mtype: "<type \'property\'>" } diff --git a/tensorflow/tools/api/golden/v2/tensorflow.train.-vocab-info.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.train.-vocab-info.pbtxt index 4ce7cb1111..39b946b82f 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.train.-vocab-info.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.train.-vocab-info.pbtxt @@ -4,6 +4,10 @@ tf_class { is_instance: "<class \'tensorflow.python.training.warm_starting_util.VocabInfo\'>" is_instance: "<type \'tuple\'>" member { + name: "axis" + mtype: "<type \'property\'>" + } + member { name: "backup_initializer" mtype: "<type \'property\'>" } |