Home
last modified time | relevance | path

Searched refs:func_graph (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/python/framework/
Dfunction_def_to_graph.py34 from tensorflow.python.framework.func_graph import FuncGraph
58 func_graph = FuncGraph(fdef.signature.name)
66 with func_graph.as_default():
76 func_graph.inputs = [
77 func_graph.get_tensor_by_name(name) for name in input_tensor_names
85 func_graph.outputs = [
86 func_graph.get_tensor_by_name(name) for name in output_tensor_names
88 func_graph.control_outputs = [
89 func_graph.get_operation_by_name(fdef.control_ret[ret_name])
93 _set_handle_data(func_graph, fdef)
[all …]
Dfunc_graph.py818 func_graph=None,
883 if func_graph is None:
884 func_graph = FuncGraph(name, collections=collections,
886 assert isinstance(func_graph, FuncGraph)
892 with func_graph.as_default(), deps_control_manager as deps_ctx:
923 func_graph.structured_input_signature = (
931 func_graph.inputs = [arg for arg in flat_func_args + flat_func_kwargs
1012 graph_variables = list(func_graph._watched_variables) # pylint: disable=protected-access
1021 resource_placeholder = func_graph.pop_capture(arg.handle)
1029 func_graph.inputs = (
[all …]
Dfunction.py945 func_graph = _FuncGraph(name, capture_by_value, allowlisted_stateful_ops,
948 with func_graph.as_default(), ops.device(device):
951 func_graph._collections = collections_ref
953 func_graph._container = container
955 func_graph._colocation_stack = colocation_stack
964 func_graph.inputs.append(argholder)
966 with vs.variable_scope("", custom_getter=func_graph.getvar):
967 outputs = func(*func_graph.inputs)
987 outputs = [func_graph.capture(t) if t.graph is not func_graph else t
989 func_graph.outputs = outputs
[all …]
Dauto_control_deps_utils.py36 def get_read_only_resource_input_indices_graph(func_graph): argument
42 for input_index, t in enumerate(func_graph.inputs):
Dauto_control_deps_test.py192 func_graph = read_var_in_while.get_concrete_function().graph
193 assert len(func_graph.inputs) == 1
197 op for op in func_graph.get_operations()
383 func_graph = write_var_in_while.get_concrete_function().graph
384 assert len(func_graph.inputs) == 1
388 op for op in func_graph.get_operations()
Dfunction_def_to_graph_test.py213 func_graph = function_def_to_graph.function_def_to_graph(fdef)
215 op = func_graph.get_operation_by_name("y")
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py28 from tensorflow.python.framework.func_graph import FuncGraph
64 def create_new_tf_function(func_graph): argument
74 func_graph.name, func_graph, func_graph.inputs, func_graph.outputs, {})
75 func.add_to_graph(func_graph.outer_graph)
76 return func_graph.name
313 func_graph = function_def_to_graph.function_def_to_graph(
315 return func_graph
Dcond_v2.py35 from tensorflow.python.framework import func_graph as func_graph_module
86 func_graph=util.CondBranchFuncGraph(
93 func_graph=util.CondBranchFuncGraph(
327 func_graph = None
329 func_graph = getattr(op, cached_attr_name, None)
331 if func_graph is None:
333 func_graph = util.get_func_graph(op, input_shapes, name_attr_list.name)
334 for external_t, internal_t in zip(inputs, func_graph.inputs):
336 func_graph.reset_captures(zip(inputs, func_graph.inputs))
338 func_graph._forward_cond = op
[all …]
Dcontrol_flow_v2_func_graphs.py21 from tensorflow.python.framework import func_graph
25 class ControlFlowFuncGraph(func_graph.FuncGraph):
38 func_graph.override_func_graph_name_scope(
Dwhile_v2.py34 from tensorflow.python.framework import func_graph as func_graph_module
151 func_graph=util.WhileCondFuncGraph(
200 func_graph=util.WhileBodyFuncGraph(
406 func_graph=util.WhileCondFuncGraph(grad_cond_name))
464 def _get_intermediates(func_graph): argument
489 reverse_captures = dict((v.ref(), k) for k, v in func_graph.captures)
491 for op in func_graph.get_operations():
498 if (o is not func_graph.inputs[0] and # Loop counter.
596 func_graph = getattr(while_op, attr_graph_name, None)
597 if func_graph is None:
[all …]
Dwhile_v2_indexed_slices_rewriter.py23 from tensorflow.python.framework import func_graph
105 body_grad_graph.outputs = func_graph.flatten(
278 func_graph.flatten(old_output_slices)[0])
281 graph.outputs = func_graph.flatten(
Dgradients_util.py36 from tensorflow.python.framework.func_graph import FuncGraph
388 def _Captures(func_graph): argument
389 if isinstance(func_graph, FuncGraph):
390 return func_graph.captures
392 assert isinstance(func_graph, framework_function._FuncGraph) # pylint: disable=protected-access
393 return func_graph.captures
Dop_selector_test.py21 from tensorflow.python.framework import func_graph
76 inner = func_graph.FuncGraph("inner")
Dscript_ops.py34 from tensorflow.python.framework import func_graph
327 elif isinstance(graph, func_graph.FuncGraph):
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py29 from tensorflow.python.framework import func_graph as func_graph_lib
358 func_graph = function_def_lib.function_def_to_graph(copy)
359 _restore_gradient_functions(func_graph, renamed_functions)
362 functions[dep].add_to_graph(func_graph)
369 func = function_lib.ConcreteFunction(func_graph)
374 if any(op.type == "TRTEngineOp" for op in func_graph.get_operations()):
383 def _restore_gradient_functions(func_graph, renamed_functions): argument
385 for op in func_graph.get_operations():
Dfunction_serialization.py23 from tensorflow.python.framework import func_graph as func_graph_module
167 func_graph=outer_graph)
Dload_v1_in_v2.py29 from tensorflow.python.framework import func_graph
191 func_graph.convert_structure_to_signature(
DBUILD441 "//tensorflow/python:func_graph",
536 "//tensorflow/python:func_graph",
552 "//tensorflow/python:func_graph",
/external/tensorflow/tensorflow/python/eager/
Dlift_to_graph_test.py25 from tensorflow.python.framework import func_graph
48 g = func_graph.FuncGraph('lifted')
75 g = func_graph.FuncGraph('lifted')
Dwrap_function.py30 from tensorflow.python.framework import func_graph
333 pruned_graph = func_graph.FuncGraph(name)
457 self.graph = func_graph.FuncGraph(name, collections=collections, **kwargs)
530 func_graph.func_graph_from_py_func(
537 func_graph=self.graph)
621 func_graph.func_graph_from_py_func(
Ddef_function.py36 from tensorflow.python.framework import func_graph as func_graph_module
281 func_graph = ops.get_default_graph()
283 func_graph.inputs + func_graph.internal_captures)
464 def __init__(self, func_graph): argument
465 self.func_graph = func_graph
469 func_graph_module.dismantle_func_graph(self.func_graph)
Dlift_to_graph.py24 from tensorflow.python.framework import func_graph
299 if (isinstance(base_graph, func_graph.FuncGraph) and
300 isinstance(graph, func_graph.FuncGraph)):
Dfunction.py53 from tensorflow.python.framework import func_graph as func_graph_module
651 def __init__(self, func_graph, attrs, func_graph_deleter): argument
657 self._func_graph = func_graph
723 func_graph=backwards_graph)
879 def __init__(self, func_graph, attrs, func_graph_deleter, argument
882 self._func_graph = func_graph
887 self._num_outputs = len(func_graph.outputs)
892 self._num_inference_outputs = len(func_graph.outputs)
894 [t for t in func_graph.outputs if backprop_util.IsTrainable(t)])
1324 def __init__(self, func_graph, attrs, func_graph_deleter, argument
[all …]
/external/tensorflow/tensorflow/c/
Dc_api_function_test.cc1216 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in DefineFunction() local
1221 TF_Operation* feed = Placeholder(func_graph.get(), s.get()); in DefineFunction()
1222 TF_Operation* neg = Neg(feed, func_graph.get(), s.get()); in DefineFunction()
1226 *func = TF_GraphToFunction(func_graph.get(), name, append_hash, -1, in DefineFunction()
1250 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in TEST_F() local
1256 NodeWithPlaceholderAttrHelper(func_graph.get(), s.get(), "node1", "v1", in TEST_F()
1258 NodeWithPlaceholderAttrHelper(func_graph.get(), s.get(), "node2", "v1", in TEST_F()
1260 NodeWithPlaceholderAttrHelper(func_graph.get(), s.get(), "node3", "v2", in TEST_F()
1265 func_graph.get(), "func", /*append_hash_to_fn_name=*/false, -1, in TEST_F()
1292 std::unique_ptr<TF_Graph, decltype(&TF_DeleteGraph)> func_graph( in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc192 Graph* func_graph = fbody->graph; in PropagateConstIntoFuncAttr() local
194 ReplaceArgUsageWithConstNode(func_graph, const_input_index_to_node)); in PropagateConstIntoFuncAttr()
201 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef)); in PropagateConstIntoFuncAttr()
214 TF_RETURN_IF_ERROR(CopyAssociatedFunctions(func_graph, lookup_fld, fld)); in PropagateConstIntoFuncAttr()

123