Remove symbols from the v2 API. Included here:
tf.train.NewCheckpointReader tf.train.replica_device_setter tf.train.init_from_checkpoint tf.train.ProfilerHook tf.profiler.* Profiler's API depends on v1 style graphs, so we'll remove this pending replacement. PiperOrigin-RevId: 221359471
This commit is contained in:
parent
07a248c2c4
commit
ac79b5c1d2
@ -122,7 +122,7 @@ def _build_advisor_options(options):
|
||||
return opts
|
||||
|
||||
|
||||
@tf_export('profiler.Profiler')
|
||||
@tf_export(v1=['profiler.Profiler'])
|
||||
class Profiler(object):
|
||||
"""TensorFlow multi-step profiler.
|
||||
|
||||
@ -306,7 +306,7 @@ class Profiler(object):
|
||||
print_mdl.WriteProfile(filename)
|
||||
|
||||
|
||||
@tf_export('profiler.profile')
|
||||
@tf_export(v1=['profiler.profile'])
|
||||
def profile(graph=None,
|
||||
run_meta=None,
|
||||
op_log=None,
|
||||
@ -381,7 +381,7 @@ def profile(graph=None,
|
||||
return tfprof_node
|
||||
|
||||
|
||||
@tf_export('profiler.advise')
|
||||
@tf_export(v1=['profiler.advise'])
|
||||
def advise(graph=None, run_meta=None, options=_DEFAULT_ADVISE_OPTIONS):
|
||||
"""Auto profile and advise.
|
||||
|
||||
|
@ -23,7 +23,7 @@ from tensorflow.python.profiler import tfprof_logger
|
||||
from tensorflow.python.util.tf_export import tf_export
|
||||
|
||||
|
||||
@tf_export('profiler.ProfileOptionBuilder')
|
||||
@tf_export(v1=['profiler.ProfileOptionBuilder'])
|
||||
class ProfileOptionBuilder(object):
|
||||
# pylint: disable=line-too-long
|
||||
"""Option Builder for Profiling API.
|
||||
|
@ -49,7 +49,7 @@ _allowed_symbols.extend([
|
||||
])
|
||||
|
||||
# Export protos
|
||||
tf_export('profiler.GraphNodeProto')(GraphNodeProto)
|
||||
tf_export('profiler.MultiGraphNodeProto')(MultiGraphNodeProto)
|
||||
tf_export('profiler.AdviceProto')(AdviceProto)
|
||||
tf_export('profiler.OpLogProto')(OpLogProto)
|
||||
tf_export(v1=['profiler.GraphNodeProto'])(GraphNodeProto)
|
||||
tf_export(v1=['profiler.MultiGraphNodeProto'])(MultiGraphNodeProto)
|
||||
tf_export(v1=['profiler.AdviceProto'])(AdviceProto)
|
||||
tf_export(v1=['profiler.OpLogProto'])(OpLogProto)
|
||||
|
@ -188,7 +188,7 @@ def merge_default_with_oplog(graph, op_log=None, run_meta=None,
|
||||
return tmp_op_log
|
||||
|
||||
|
||||
@tf_export('profiler.write_op_log')
|
||||
@tf_export(v1=['profiler.write_op_log'])
|
||||
def write_op_log(graph, log_dir, op_log=None, run_meta=None, add_trace=True):
|
||||
"""Log provided 'op_log', and add additional model information below.
|
||||
|
||||
|
@ -68,7 +68,6 @@ TENSORFLOW_API_INIT_FILES = [
|
||||
"metrics/__init__.py",
|
||||
"nn/__init__.py",
|
||||
"nn/rnn_cell/__init__.py",
|
||||
"profiler/__init__.py",
|
||||
"quantization/__init__.py",
|
||||
"random/__init__.py",
|
||||
"saved_model/__init__.py",
|
||||
|
@ -976,7 +976,7 @@ class FeedFnHook(session_run_hook.SessionRunHook):
|
||||
fetches=None, feed_dict=self.feed_fn())
|
||||
|
||||
|
||||
@tf_export("train.ProfilerHook")
|
||||
@tf_export(v1=["train.ProfilerHook"])
|
||||
class ProfilerHook(session_run_hook.SessionRunHook):
|
||||
"""Captures CPU/GPU profiling information every N steps or seconds.
|
||||
|
||||
|
@ -101,7 +101,7 @@ def list_variables(ckpt_dir_or_file):
|
||||
return result
|
||||
|
||||
|
||||
@tf_export("train.init_from_checkpoint")
|
||||
@tf_export(v1=["train.init_from_checkpoint"])
|
||||
def init_from_checkpoint(ckpt_dir_or_file, assignment_map):
|
||||
"""Replaces `tf.Variable` initializers so they load from a checkpoint file.
|
||||
|
||||
|
@ -130,7 +130,7 @@ class _ReplicaDeviceChooser(object):
|
||||
return worker_device.to_string()
|
||||
|
||||
|
||||
@tf_export("train.replica_device_setter")
|
||||
@tf_export(v1=["train.replica_device_setter"])
|
||||
def replica_device_setter(ps_tasks=0, ps_device="/job:ps",
|
||||
worker_device="/job:worker", merge_devices=True,
|
||||
cluster=None, ps_ops=None, ps_strategy=None):
|
||||
|
@ -165,7 +165,6 @@ def NewCheckpointReader(filepattern):
|
||||
from tensorflow.python.util import compat
|
||||
return CheckpointReader(compat.as_bytes(filepattern), status)
|
||||
|
||||
NewCheckpointReader._tf_api_names = ['train.NewCheckpointReader']
|
||||
NewCheckpointReader._tf_api_names_v1 = ['train.NewCheckpointReader']
|
||||
%}
|
||||
|
||||
|
@ -328,10 +328,6 @@ tf_module {
|
||||
name: "ones_initializer"
|
||||
mtype: "<type \'type\'>"
|
||||
}
|
||||
member {
|
||||
name: "profiler"
|
||||
mtype: "<type \'module\'>"
|
||||
}
|
||||
member {
|
||||
name: "pywrap_tensorflow"
|
||||
mtype: "<type \'module\'>"
|
||||
|
@ -1,12 +0,0 @@
|
||||
path: "tensorflow.profiler.AdviceProto.Checker"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "Checker"
|
||||
field {
|
||||
name: "reports"
|
||||
number: 2
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_STRING
|
||||
}
|
||||
}
|
||||
}
|
@ -1,22 +0,0 @@
|
||||
path: "tensorflow.profiler.AdviceProto.CheckersEntry"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "CheckersEntry"
|
||||
field {
|
||||
name: "key"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_STRING
|
||||
}
|
||||
field {
|
||||
name: "value"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.AdviceProto.Checker"
|
||||
}
|
||||
options {
|
||||
map_entry: true
|
||||
}
|
||||
}
|
||||
}
|
@ -1,41 +0,0 @@
|
||||
path: "tensorflow.profiler.AdviceProto"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "AdviceProto"
|
||||
field {
|
||||
name: "checkers"
|
||||
number: 1
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.AdviceProto.CheckersEntry"
|
||||
}
|
||||
nested_type {
|
||||
name: "CheckersEntry"
|
||||
field {
|
||||
name: "key"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_STRING
|
||||
}
|
||||
field {
|
||||
name: "value"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.AdviceProto.Checker"
|
||||
}
|
||||
options {
|
||||
map_entry: true
|
||||
}
|
||||
}
|
||||
nested_type {
|
||||
name: "Checker"
|
||||
field {
|
||||
name: "reports"
|
||||
number: 2
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_STRING
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -1,22 +0,0 @@
|
||||
path: "tensorflow.profiler.GraphNodeProto.InputShapesEntry"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "InputShapesEntry"
|
||||
field {
|
||||
name: "key"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT32
|
||||
}
|
||||
field {
|
||||
name: "value"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.TensorShapeProto"
|
||||
}
|
||||
options {
|
||||
map_entry: true
|
||||
}
|
||||
}
|
||||
}
|
@ -1,191 +0,0 @@
|
||||
path: "tensorflow.profiler.GraphNodeProto"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "GraphNodeProto"
|
||||
field {
|
||||
name: "name"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_STRING
|
||||
}
|
||||
field {
|
||||
name: "tensor_value"
|
||||
number: 15
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.TFProfTensorProto"
|
||||
}
|
||||
field {
|
||||
name: "run_count"
|
||||
number: 21
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "exec_micros"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "accelerator_exec_micros"
|
||||
number: 17
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "cpu_exec_micros"
|
||||
number: 18
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "requested_bytes"
|
||||
number: 3
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "peak_bytes"
|
||||
number: 24
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "residual_bytes"
|
||||
number: 25
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "output_bytes"
|
||||
number: 26
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "parameters"
|
||||
number: 4
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "float_ops"
|
||||
number: 13
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "devices"
|
||||
number: 10
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_STRING
|
||||
}
|
||||
field {
|
||||
name: "total_definition_count"
|
||||
number: 23
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_run_count"
|
||||
number: 22
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_exec_micros"
|
||||
number: 6
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_accelerator_exec_micros"
|
||||
number: 19
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_cpu_exec_micros"
|
||||
number: 20
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_requested_bytes"
|
||||
number: 7
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_peak_bytes"
|
||||
number: 27
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_residual_bytes"
|
||||
number: 28
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_output_bytes"
|
||||
number: 29
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_parameters"
|
||||
number: 8
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_float_ops"
|
||||
number: 14
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "shapes"
|
||||
number: 11
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.TensorShapeProto"
|
||||
}
|
||||
field {
|
||||
name: "input_shapes"
|
||||
number: 16
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.GraphNodeProto.InputShapesEntry"
|
||||
}
|
||||
field {
|
||||
name: "children"
|
||||
number: 12
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.GraphNodeProto"
|
||||
}
|
||||
nested_type {
|
||||
name: "InputShapesEntry"
|
||||
field {
|
||||
name: "key"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT32
|
||||
}
|
||||
field {
|
||||
name: "value"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.TensorShapeProto"
|
||||
}
|
||||
options {
|
||||
map_entry: true
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -1,134 +0,0 @@
|
||||
path: "tensorflow.profiler.MultiGraphNodeProto"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "MultiGraphNodeProto"
|
||||
field {
|
||||
name: "name"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_STRING
|
||||
}
|
||||
field {
|
||||
name: "exec_micros"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "accelerator_exec_micros"
|
||||
number: 12
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "cpu_exec_micros"
|
||||
number: 13
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "requested_bytes"
|
||||
number: 3
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "peak_bytes"
|
||||
number: 16
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "residual_bytes"
|
||||
number: 17
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "output_bytes"
|
||||
number: 18
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "parameters"
|
||||
number: 4
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "float_ops"
|
||||
number: 5
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_exec_micros"
|
||||
number: 6
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_accelerator_exec_micros"
|
||||
number: 14
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_cpu_exec_micros"
|
||||
number: 15
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_requested_bytes"
|
||||
number: 7
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_peak_bytes"
|
||||
number: 19
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_residual_bytes"
|
||||
number: 20
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_output_bytes"
|
||||
number: 21
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_parameters"
|
||||
number: 8
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "total_float_ops"
|
||||
number: 9
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "graph_nodes"
|
||||
number: 10
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.GraphNodeProto"
|
||||
}
|
||||
field {
|
||||
name: "children"
|
||||
number: 11
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.MultiGraphNodeProto"
|
||||
}
|
||||
}
|
||||
}
|
@ -1,21 +0,0 @@
|
||||
path: "tensorflow.profiler.OpLogProto.IdToStringEntry"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "IdToStringEntry"
|
||||
field {
|
||||
name: "key"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "value"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_STRING
|
||||
}
|
||||
options {
|
||||
map_entry: true
|
||||
}
|
||||
}
|
||||
}
|
@ -1,38 +0,0 @@
|
||||
path: "tensorflow.profiler.OpLogProto"
|
||||
tf_proto {
|
||||
descriptor {
|
||||
name: "OpLogProto"
|
||||
field {
|
||||
name: "log_entries"
|
||||
number: 1
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.OpLogEntry"
|
||||
}
|
||||
field {
|
||||
name: "id_to_string"
|
||||
number: 2
|
||||
label: LABEL_REPEATED
|
||||
type: TYPE_MESSAGE
|
||||
type_name: ".tensorflow.tfprof.OpLogProto.IdToStringEntry"
|
||||
}
|
||||
nested_type {
|
||||
name: "IdToStringEntry"
|
||||
field {
|
||||
name: "key"
|
||||
number: 1
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_INT64
|
||||
}
|
||||
field {
|
||||
name: "value"
|
||||
number: 2
|
||||
label: LABEL_OPTIONAL
|
||||
type: TYPE_STRING
|
||||
}
|
||||
options {
|
||||
map_entry: true
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
@ -1,93 +0,0 @@
|
||||
path: "tensorflow.profiler.ProfileOptionBuilder"
|
||||
tf_class {
|
||||
is_instance: "<class \'tensorflow.python.profiler.option_builder.ProfileOptionBuilder\'>"
|
||||
is_instance: "<type \'object\'>"
|
||||
member_method {
|
||||
name: "__init__"
|
||||
argspec: "args=[\'self\', \'options\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "account_displayed_op_only"
|
||||
argspec: "args=[\'self\', \'is_true\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "build"
|
||||
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "float_operation"
|
||||
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "order_by"
|
||||
argspec: "args=[\'self\', \'attribute\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "select"
|
||||
argspec: "args=[\'self\', \'attributes\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "time_and_memory"
|
||||
argspec: "args=[\'min_micros\', \'min_bytes\', \'min_accelerator_micros\', \'min_cpu_micros\', \'min_peak_bytes\', \'min_residual_bytes\', \'min_output_bytes\'], varargs=None, keywords=None, defaults=[\'1\', \'1\', \'0\', \'0\', \'0\', \'0\', \'0\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "trainable_variables_parameter"
|
||||
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_accounted_types"
|
||||
argspec: "args=[\'self\', \'account_type_regexes\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_empty_output"
|
||||
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_file_output"
|
||||
argspec: "args=[\'self\', \'outfile\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_max_depth"
|
||||
argspec: "args=[\'self\', \'max_depth\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_min_execution_time"
|
||||
argspec: "args=[\'self\', \'min_micros\', \'min_accelerator_micros\', \'min_cpu_micros\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'0\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "with_min_float_operations"
|
||||
argspec: "args=[\'self\', \'min_float_ops\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_min_memory"
|
||||
argspec: "args=[\'self\', \'min_bytes\', \'min_peak_bytes\', \'min_residual_bytes\', \'min_output_bytes\'], varargs=None, keywords=None, defaults=[\'0\', \'0\', \'0\', \'0\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "with_min_occurrence"
|
||||
argspec: "args=[\'self\', \'min_occurrence\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_min_parameters"
|
||||
argspec: "args=[\'self\', \'min_params\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_node_names"
|
||||
argspec: "args=[\'self\', \'start_name_regexes\', \'show_name_regexes\', \'hide_name_regexes\', \'trim_name_regexes\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "with_pprof_output"
|
||||
argspec: "args=[\'self\', \'pprof_file\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_stdout_output"
|
||||
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_step"
|
||||
argspec: "args=[\'self\', \'step\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "with_timeline_output"
|
||||
argspec: "args=[\'self\', \'timeline_file\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
}
|
@ -1,37 +0,0 @@
|
||||
path: "tensorflow.profiler.Profiler"
|
||||
tf_class {
|
||||
is_instance: "<class \'tensorflow.python.profiler.model_analyzer.Profiler\'>"
|
||||
is_instance: "<type \'object\'>"
|
||||
member_method {
|
||||
name: "__init__"
|
||||
argspec: "args=[\'self\', \'graph\', \'op_log\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "add_step"
|
||||
argspec: "args=[\'self\', \'step\', \'run_meta\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "advise"
|
||||
argspec: "args=[\'self\', \'options\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "profile_graph"
|
||||
argspec: "args=[\'self\', \'options\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "profile_name_scope"
|
||||
argspec: "args=[\'self\', \'options\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "profile_operations"
|
||||
argspec: "args=[\'self\', \'options\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "profile_python"
|
||||
argspec: "args=[\'self\', \'options\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "serialize_to_string"
|
||||
argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
}
|
@ -1,39 +0,0 @@
|
||||
path: "tensorflow.profiler"
|
||||
tf_module {
|
||||
member {
|
||||
name: "AdviceProto"
|
||||
mtype: "<class \'google.protobuf.pyext.cpp_message.GeneratedProtocolMessageType\'>"
|
||||
}
|
||||
member {
|
||||
name: "GraphNodeProto"
|
||||
mtype: "<class \'google.protobuf.pyext.cpp_message.GeneratedProtocolMessageType\'>"
|
||||
}
|
||||
member {
|
||||
name: "MultiGraphNodeProto"
|
||||
mtype: "<class \'google.protobuf.pyext.cpp_message.GeneratedProtocolMessageType\'>"
|
||||
}
|
||||
member {
|
||||
name: "OpLogProto"
|
||||
mtype: "<class \'google.protobuf.pyext.cpp_message.GeneratedProtocolMessageType\'>"
|
||||
}
|
||||
member {
|
||||
name: "ProfileOptionBuilder"
|
||||
mtype: "<type \'type\'>"
|
||||
}
|
||||
member {
|
||||
name: "Profiler"
|
||||
mtype: "<type \'type\'>"
|
||||
}
|
||||
member_method {
|
||||
name: "advise"
|
||||
argspec: "args=[\'graph\', \'run_meta\', \'options\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'0\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "profile"
|
||||
argspec: "args=[\'graph\', \'run_meta\', \'op_log\', \'cmd\', \'options\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'scope\', \'0\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "write_op_log"
|
||||
argspec: "args=[\'graph\', \'log_dir\', \'op_log\', \'run_meta\', \'add_trace\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'True\'], "
|
||||
}
|
||||
}
|
@ -1,30 +0,0 @@
|
||||
path: "tensorflow.train.ProfilerHook"
|
||||
tf_class {
|
||||
is_instance: "<class \'tensorflow.python.training.basic_session_run_hooks.ProfilerHook\'>"
|
||||
is_instance: "<class \'tensorflow.python.training.session_run_hook.SessionRunHook\'>"
|
||||
is_instance: "<type \'object\'>"
|
||||
member_method {
|
||||
name: "__init__"
|
||||
argspec: "args=[\'self\', \'save_steps\', \'save_secs\', \'output_dir\', \'show_dataflow\', \'show_memory\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'\', \'True\', \'False\'], "
|
||||
}
|
||||
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"
|
||||
}
|
||||
}
|
@ -136,10 +136,6 @@ tf_module {
|
||||
name: "Optimizer"
|
||||
mtype: "<type \'type\'>"
|
||||
}
|
||||
member {
|
||||
name: "ProfilerHook"
|
||||
mtype: "<type \'type\'>"
|
||||
}
|
||||
member {
|
||||
name: "ProximalAdagradOptimizer"
|
||||
mtype: "<type \'type\'>"
|
||||
@ -224,10 +220,6 @@ tf_module {
|
||||
name: "WorkerSessionCreator"
|
||||
mtype: "<type \'type\'>"
|
||||
}
|
||||
member_method {
|
||||
name: "NewCheckpointReader"
|
||||
argspec: "args=[\'filepattern\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "cosine_decay"
|
||||
argspec: "args=[\'learning_rate\', \'global_step\', \'decay_steps\', \'alpha\', \'name\'], varargs=None, keywords=None, defaults=[\'0.0\', \'None\'], "
|
||||
@ -244,10 +236,6 @@ tf_module {
|
||||
name: "get_checkpoint_state"
|
||||
argspec: "args=[\'checkpoint_dir\', \'latest_filename\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "init_from_checkpoint"
|
||||
argspec: "args=[\'ckpt_dir_or_file\', \'assignment_map\'], varargs=None, keywords=None, defaults=None"
|
||||
}
|
||||
member_method {
|
||||
name: "inverse_time_decay"
|
||||
argspec: "args=[\'learning_rate\', \'global_step\', \'decay_steps\', \'decay_rate\', \'staircase\', \'name\'], varargs=None, keywords=None, defaults=[\'False\', \'None\'], "
|
||||
@ -288,10 +276,6 @@ tf_module {
|
||||
name: "polynomial_decay"
|
||||
argspec: "args=[\'learning_rate\', \'global_step\', \'decay_steps\', \'end_learning_rate\', \'power\', \'cycle\', \'name\'], varargs=None, keywords=None, defaults=[\'0.0001\', \'1.0\', \'False\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "replica_device_setter"
|
||||
argspec: "args=[\'ps_tasks\', \'ps_device\', \'worker_device\', \'merge_devices\', \'cluster\', \'ps_ops\', \'ps_strategy\'], varargs=None, keywords=None, defaults=[\'0\', \'/job:ps\', \'/job:worker\', \'True\', \'None\', \'None\', \'None\'], "
|
||||
}
|
||||
member_method {
|
||||
name: "sdca_fprint"
|
||||
argspec: "args=[\'input\', \'name\'], varargs=None, keywords=None, defaults=[\'None\'], "
|
||||
|
Loading…
x
Reference in New Issue
Block a user