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

Export tf.train.* session_run_hook.py classes to tf.estimator.* (exporting to both v1 and v2). Keep the existing only in v1.

Browse Source 
PiperOrigin-RevId: 225276892
This commit is contained in:
Rick Chao 2018-12-12 16:29:27 -08:00 committed by TensorFlower Gardener
parent 090c5ed073
commit 2885c4bb67
13 changed files with 224 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/training/session_run_hook.py
View File

@ -94,7 +94,7 @@ import collections
from tensorflow.python.util.tf_export import tf_export
@tf_export("train.SessionRunHook")
@tf_export(v1=["train.SessionRunHook"])
class SessionRunHook(object):
"""Hook to extend calls to MonitoredSession.run()."""

27
tensorflow/tools/api/golden/v1/tensorflow.estimator.-session-run-args.pbtxt Normal file
View File

@ -0,0 +1,27 @@
path: "tensorflow.estimator.SessionRunArgs"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunArgs\'>"
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunArgs\'>"
is_instance: "<type \'tuple\'>"
member {
name: "feed_dict"
mtype: "<type \'property\'>"
}
member {
name: "fetches"
mtype: "<type \'property\'>"
}
member {
name: "options"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
}
member_method {
name: "count"
}
member_method {
name: "index"
}
}

25
tensorflow/tools/api/golden/v1/tensorflow.estimator.-session-run-context.pbtxt Normal file
View File

@ -0,0 +1,25 @@
path: "tensorflow.estimator.SessionRunContext"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunContext\'>"
is_instance: "<type \'object\'>"
member {
name: "original_args"
mtype: "<type \'property\'>"
}
member {
name: "session"
mtype: "<type \'property\'>"
}
member {
name: "stop_requested"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
argspec: "args=[\'self\', \'original_args\', \'session\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "request_stop"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
}

2
tensorflow/tools/api/golden/v2/tensorflow.train.-session-run-hook.pbtxt → tensorflow/tools/api/golden/v1/tensorflow.estimator.-session-run-hook.pbtxt
View File

@ -1,4 +1,4 @@
path: "tensorflow.train.SessionRunHook"
path: "tensorflow.estimator.SessionRunHook"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunHook\'>"
is_instance: "<type \'object\'>"

27
tensorflow/tools/api/golden/v1/tensorflow.estimator.-session-run-values.pbtxt Normal file
View File

@ -0,0 +1,27 @@
path: "tensorflow.estimator.SessionRunValues"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunValues\'>"
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunValues\'>"
is_instance: "<type \'tuple\'>"
member {
name: "options"
mtype: "<type \'property\'>"
}
member {
name: "results"
mtype: "<type \'property\'>"
}
member {
name: "run_metadata"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
}
member_method {
name: "count"
}
member_method {
name: "index"
}
}

16
tensorflow/tools/api/golden/v1/tensorflow.estimator.pbtxt
View File

@ -132,6 +132,22 @@ tf_module {
name: "SecondOrStepTimer"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunArgs"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunContext"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunHook"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunValues"
mtype: "<type \'type\'>"
}
member {
name: "StepCounterHook"
mtype: "<type \'type\'>"

27
tensorflow/tools/api/golden/v2/tensorflow.estimator.-session-run-args.pbtxt Normal file
View File

@ -0,0 +1,27 @@
path: "tensorflow.estimator.SessionRunArgs"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunArgs\'>"
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunArgs\'>"
is_instance: "<type \'tuple\'>"
member {
name: "feed_dict"
mtype: "<type \'property\'>"
}
member {
name: "fetches"
mtype: "<type \'property\'>"
}
member {
name: "options"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
}
member_method {
name: "count"
}
member_method {
name: "index"
}
}

25
tensorflow/tools/api/golden/v2/tensorflow.estimator.-session-run-context.pbtxt Normal file
View File

@ -0,0 +1,25 @@
path: "tensorflow.estimator.SessionRunContext"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunContext\'>"
is_instance: "<type \'object\'>"
member {
name: "original_args"
mtype: "<type \'property\'>"
}
member {
name: "session"
mtype: "<type \'property\'>"
}
member {
name: "stop_requested"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
argspec: "args=[\'self\', \'original_args\', \'session\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "request_stop"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
}

28
tensorflow/tools/api/golden/v2/tensorflow.estimator.-session-run-hook.pbtxt Normal file
View File

@ -0,0 +1,28 @@
path: "tensorflow.estimator.SessionRunHook"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunHook\'>"
is_instance: "<type \'object\'>"
member_method {
name: "__init__"
}
member_method {
name: "after_create_session"
argspec: "args=[\'self\', \'session\', \'coord\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "after_run"
argspec: "args=[\'self\', \'run_context\', \'run_values\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "before_run"
argspec: "args=[\'self\', \'run_context\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "begin"
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
}
member_method {
name: "end"
argspec: "args=[\'self\', \'session\'], varargs=None, keywords=None, defaults=None"
}
}

27
tensorflow/tools/api/golden/v2/tensorflow.estimator.-session-run-values.pbtxt Normal file
View File

@ -0,0 +1,27 @@
path: "tensorflow.estimator.SessionRunValues"
tf_class {
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunValues\'>"
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunValues\'>"
is_instance: "<type \'tuple\'>"
member {
name: "options"
mtype: "<type \'property\'>"
}
member {
name: "results"
mtype: "<type \'property\'>"
}
member {
name: "run_metadata"
mtype: "<type \'property\'>"
}
member_method {
name: "__init__"
}
member_method {
name: "count"
}
member_method {
name: "index"
}
}

16
tensorflow/tools/api/golden/v2/tensorflow.estimator.pbtxt
View File

@ -132,6 +132,22 @@ tf_module {
name: "SecondOrStepTimer"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunArgs"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunContext"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunHook"
mtype: "<type \'type\'>"
}
member {
name: "SessionRunValues"
mtype: "<type \'type\'>"
}
member {
name: "StepCounterHook"
mtype: "<type \'type\'>"

4
tensorflow/tools/api/golden/v2/tensorflow.train.pbtxt
View File

@ -72,10 +72,6 @@ tf_module {
name: "ServerDef"
mtype: "<class \'google.protobuf.pyext.cpp_message.GeneratedProtocolMessageType\'>"
}
member {
name: "SessionRunHook"
mtype: "<type \'type\'>"
}
member_method {
name: "cosine_decay"
argspec: "args=[\'learning_rate\', \'global_step\', \'decay_steps\', \'alpha\', \'name\'], varargs=None, keywords=None, defaults=[\'0.0\', \'None\'], "

7
tensorflow/tools/compatibility/renames_v2.py
View File

@ -625,9 +625,10 @@ renames = {
'tf.train.Server': 'tf.distribute.Server',
'tf.train.SessionCreator': 'tf.compat.v1.train.SessionCreator',
'tf.train.SessionManager': 'tf.compat.v1.train.SessionManager',
'tf.train.SessionRunArgs': 'tf.compat.v1.train.SessionRunArgs',
'tf.train.SessionRunContext': 'tf.compat.v1.train.SessionRunContext',
'tf.train.SessionRunValues': 'tf.compat.v1.train.SessionRunValues',
'tf.train.SessionRunArgs': 'tf.estimator.SessionRunArgs',
'tf.train.SessionRunContext': 'tf.estimator.SessionRunContext',
'tf.train.SessionRunHook': 'tf.estimator.SessionRunHook',
'tf.train.SessionRunValues': 'tf.estimator.SessionRunValues',
'tf.train.SingularMonitoredSession': 'tf.compat.v1.train.SingularMonitoredSession',
'tf.train.StepCounterHook': 'tf.estimator.StepCounterHook',
'tf.train.StopAtStepHook': 'tf.estimator.StopAtStepHook',