diff --git a/tensorflow/python/framework/func_graph.py b/tensorflow/python/framework/func_graph.py index f70bd75e36b..6d5a303977f 100644 --- a/tensorflow/python/framework/func_graph.py +++ b/tensorflow/python/framework/func_graph.py @@ -49,6 +49,7 @@ from tensorflow.python.util import nest from tensorflow.python.util import object_identity from tensorflow.python.util import tf_contextlib from tensorflow.python.util import tf_decorator +from tensorflow.python.util.tf_export import tf_export ALLOWLIST_COLLECTIONS = [ ops.GraphKeys.GLOBAL_VARIABLES, @@ -133,6 +134,7 @@ def convert_structure_to_signature(structure, arg_names=None): return nest.pack_sequence_as(structure, mapped) +@tf_export("__internal__.FuncGraph", v1=[]) class FuncGraph(ops.Graph): """Graph representing a function body. diff --git a/tensorflow/tools/api/golden/v2/tensorflow.__internal__.-func-graph.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.__internal__.-func-graph.pbtxt new file mode 100644 index 00000000000..d7616f025dc --- /dev/null +++ b/tensorflow/tools/api/golden/v2/tensorflow.__internal__.-func-graph.pbtxt @@ -0,0 +1,242 @@ +path: "tensorflow.__internal__.FuncGraph" +tf_class { + is_instance: "" + is_instance: "" + is_instance: "" + member { + name: "building_function" + mtype: "" + } + member { + name: "captures" + mtype: "" + } + member { + name: "collections" + mtype: "" + } + member { + name: "deferred_external_captures" + mtype: "" + } + member { + name: "deferred_internal_captures" + mtype: "" + } + member { + name: "external_captures" + mtype: "" + } + member { + name: "finalized" + mtype: "" + } + member { + name: "graph_def_versions" + mtype: "" + } + member { + name: "internal_captures" + mtype: "" + } + member { + name: "outer_graph" + mtype: "" + } + member { + name: "output_shapes" + mtype: "" + } + member { + name: "output_types" + mtype: "" + } + member { + name: "saveable" + mtype: "" + } + member { + name: "saving_errors" + mtype: "" + } + member { + name: "seed" + mtype: "" + } + member { + name: "trainable_variables" + mtype: "" + } + member { + name: "variable_captures" + mtype: "" + } + member { + name: "variables" + mtype: "" + } + member { + name: "version" + mtype: "" + } + member_method { + name: "__init__" + argspec: "args=[\'self\', \'name\', \'collections\', \'capture_by_value\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "add_capture" + argspec: "args=[\'self\', \'tensor\', \'placeholder\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "add_to_collection" + argspec: "args=[\'self\', \'name\', \'value\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "add_to_collections" + argspec: "args=[\'self\', \'names\', \'value\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "as_default" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "as_graph_def" + argspec: "args=[\'self\', \'from_version\', \'add_shapes\'], varargs=None, keywords=None, defaults=[\'None\', \'False\'], " + } + member_method { + name: "as_graph_element" + argspec: "args=[\'self\', \'obj\', \'allow_tensor\', \'allow_operation\'], varargs=None, keywords=None, defaults=[\'True\', \'True\'], " + } + member_method { + name: "capture" + argspec: "args=[\'self\', \'tensor\', \'name\', \'shape\'], varargs=None, keywords=None, defaults=[\'None\', \'None\'], " + } + member_method { + name: "capture_call_time_value" + argspec: "args=[\'self\', \'closure\', \'spec\', \'key\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "capture_distributed_variable" + argspec: "args=[\'self\', \'variable\', \'placeholder\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "capture_eager_tensor" + argspec: "args=[\'self\', \'tensor\', \'name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "captured" + argspec: "args=[\'self\', \'tensor\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "clear_captures" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "clear_collection" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "colocate_with" + argspec: "args=[\'self\', \'op\', \'ignore_existing\'], varargs=None, keywords=None, defaults=[\'False\'], " + } + member_method { + name: "container" + argspec: "args=[\'self\', \'container_name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "control_dependencies" + argspec: "args=[\'self\', \'control_inputs\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "create_op" + argspec: "args=[\'self\', \'op_type\', \'inputs\', \'dtypes\', \'input_types\', \'name\', \'attrs\', \'op_def\', \'compute_shapes\', \'compute_device\'], varargs=None, keywords=None, defaults=[\'None\', \'None\', \'None\', \'None\', \'None\', \'True\', \'True\'], " + } + member_method { + name: "device" + argspec: "args=[\'self\', \'device_name_or_function\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "finalize" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "get_all_collection_keys" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "get_collection" + argspec: "args=[\'self\', \'name\', \'scope\'], varargs=None, keywords=None, defaults=[\'None\'], " + } + member_method { + name: "get_collection_ref" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "get_name_scope" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "get_operation_by_name" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "get_operations" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "get_tensor_by_name" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "gradient_override_map" + argspec: "args=[\'self\', \'op_type_map\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "is_feedable" + argspec: "args=[\'self\', \'tensor\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "is_fetchable" + argspec: "args=[\'self\', \'tensor_or_op\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "mark_as_unsaveable" + argspec: "args=[\'self\', \'error_message\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "name_scope" + argspec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "pop_capture" + argspec: "args=[\'self\', \'tensor\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "prevent_feeding" + argspec: "args=[\'self\', \'tensor\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "prevent_fetching" + argspec: "args=[\'self\', \'op\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "replace_capture" + argspec: "args=[\'self\', \'tensor\', \'placeholder\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "reset_captures" + argspec: "args=[\'self\', \'capture_list\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "switch_to_thread_local" + argspec: "args=[\'self\'], varargs=None, keywords=None, defaults=None" + } + member_method { + name: "unique_name" + argspec: "args=[\'self\', \'name\', \'mark_as_used\'], varargs=None, keywords=None, defaults=[\'True\'], " + } + member_method { + name: "watch_variable" + argspec: "args=[\'self\', \'v\'], varargs=None, keywords=None, defaults=None" + } +} diff --git a/tensorflow/tools/api/golden/v2/tensorflow.__internal__.pbtxt b/tensorflow/tools/api/golden/v2/tensorflow.__internal__.pbtxt index 4b9c7a0c761..19279b00517 100644 --- a/tensorflow/tools/api/golden/v2/tensorflow.__internal__.pbtxt +++ b/tensorflow/tools/api/golden/v2/tensorflow.__internal__.pbtxt @@ -4,6 +4,10 @@ tf_module { name: "CompositeTensor" mtype: "" } + member { + name: "FuncGraph" + mtype: "" + } member { name: "decorator" mtype: ""