| /external/tensorflow/tensorflow/python/ops/ |
| control_flow_util_v2.py | 26 from tensorflow.python.framework.func_graph import FuncGraph
65 def create_new_tf_function(func_graph):
66 """Converts func_graph to a TF_Function and adds it to the current graph.
69 func_graph: FuncGraph
75 func_graph.name, func_graph, func_graph.inputs, func_graph.outputs, {})
76 func.add_to_graph(func_graph.outer_graph)
77 return func_graph.nam
[all...] |
| cond_v2.py | 29 from tensorflow.python.framework import func_graph as func_graph_module
71 func_graph=util.CondBranchFuncGraph(
78 func_graph=util.CondBranchFuncGraph(
243 # from inside a Defun. We build the `func_graph` with `if_op.graph` as its
246 # in `func_graph` from its gradient graph in `_resolve_grad_inputs`.
248 func_graph = function_def_to_graph.function_def_to_graph(
250 func_graph.captures = collections.OrderedDict(zip(inputs,
251 func_graph.inputs))
253 func_graph._if = if_op
254 return func_graph
[all...] |
| while_v2_indexed_slices_rewriter.py | 23 from tensorflow.python.framework import func_graph
96 body_grad_graph.outputs = func_graph.flatten(
266 flat_idx = graph.outputs.index(func_graph.flatten(old_output_slices)[0])
269 graph.outputs = func_graph.flatten(
|
| while_v2.py | 28 from tensorflow.python.framework import func_graph as func_graph_module
127 func_graph=util.WhileCondFuncGraph(
171 func_graph=util.WhileBodyFuncGraph(
312 func_graph=util.WhileCondFuncGraph(grad_cond_name))
414 # `func_graph` with `while_op.graph` as its `outer_graph`. This resembles how
418 func_graph = function_def_to_graph.function_def_to_graph(fdef, input_shapes)
419 func_graph._while = while_op
420 return func_graph
425 """Builds and returns the gradient FuncGraph of `func_graph` and its args.
456 func_graph=_WhileBodyGradFuncGraph(name, cond_graph, body_graph
[all...] |
| gradients_util.py | 36 from tensorflow.python.framework.func_graph import FuncGraph
454 def _Captures(func_graph):
455 if isinstance(func_graph, FuncGraph):
456 return func_graph.captures
458 assert isinstance(func_graph, framework_function._FuncGraph) # pylint: disable=protected-access
459 return func_graph._captured # pylint: disable=protected-access
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| function_def_to_graph.py | 28 from tensorflow.python.framework.func_graph import FuncGraph
50 func_graph = FuncGraph(fdef.signature.name)
54 with func_graph.as_default():
64 func_graph.inputs = [
65 func_graph.get_tensor_by_name(name) for name in input_tensor_names
73 func_graph.outputs = [
74 func_graph.get_tensor_by_name(name) for name in output_tensor_names
76 func_graph.control_outputs = [
77 func_graph.get_operation_by_name(fdef.control_ret[ret_name])
81 return func_graph
[all...] |
| func_graph.py | 47 # function -> func_graph
520 func_graph=None,
543 func_graph: Optional. An instance of FuncGraph. If provided, we will use
585 if func_graph is None:
586 func_graph = FuncGraph(name, collections=collections,
588 assert isinstance(func_graph, FuncGraph)
593 with func_graph.as_default(), control_manager as a:
624 func_graph.structured_input_signature = (
632 func_graph.inputs = [arg for arg in flat_func_args + flat_func_kwargs
714 resource_placeholder = func_graph.captures.pop(arg.handle, None
[all...] |
| function_def_to_graph_test.py | 217 func_graph = function_def_to_graph.function_def_to_graph(fdef)
218 with func_graph.as_default():
219 x_ph, y_ph = func_graph.inputs
220 with self.session(graph=func_graph) as sess:
222 sess.run(func_graph.outputs[0], feed_dict={
243 func_graph = function_def_to_graph.function_def_to_graph(fdef)
245 op = func_graph.get_operation_by_name("y")
|
| function.py | 877 func_graph = _FuncGraph(name, capture_by_value, whitelisted_stateful_ops,
880 with func_graph.as_default(), ops.device(device):
883 func_graph._collections = collections_ref
885 func_graph._container = container
887 func_graph._colocation_stack = colocation_stack
[all...] |
| convert_to_constants.py | 25 from tensorflow.python.framework import func_graph
91 output_graph = func_graph.FuncGraph(input_func.graph.name)
|
| /external/tensorflow/tensorflow/python/eager/ |
| lift_to_graph_test.py | 24 from tensorflow.python.framework import func_graph
44 g = func_graph.FuncGraph('lifted')
|
| wrap_function.py | 27 from tensorflow.python.framework import func_graph
176 pruned_graph = func_graph.FuncGraph(name)
263 self.graph = func_graph.FuncGraph(name, collections=collections, **kwargs)
278 func_graph.func_graph_from_py_func(
283 func_graph=self.graph)
286 # If an existing FuncGraph is passed into the `func_graph` arg, the inputs
360 func_graph.func_graph_from_py_func(
|
| def_function.py | 28 from tensorflow.python.framework import func_graph as func_graph_module
159 func_graph = ops.get_default_graph()
161 func_graph.inputs + func_graph.internal_captures)
238 def __init__(self, func_graph):
239 self.func_graph = func_graph
243 func_graph_module.dismantle_func_graph(self.func_graph)
[all...] |
| lift_to_graph.py | 25 from tensorflow.python.framework import func_graph
280 if (isinstance(base_graph, func_graph.FuncGraph) and
281 isinstance(graph, func_graph.FuncGraph)):
|
| function.py | 43 from tensorflow.python.framework import func_graph as func_graph_module
450 def __init__(self, func_graph, attrs=None, signature=None):
454 func_graph: An instance of FuncGraph: the function body to wrap.
467 self._func_graph = func_graph
[all...] |
| /external/tensorflow/tensorflow/python/saved_model/ |
| function_deserialization.py | 27 from tensorflow.python.framework import func_graph as func_graph_lib
246 func_graph = function_def_lib.function_def_to_graph(copy)
248 functions[dep].add_to_graph(func_graph)
249 func = function_lib.ConcreteFunction(func_graph)
|
| function_serialization.py | 22 from tensorflow.python.framework import func_graph as func_graph_module
|
| /external/tensorflow/tensorflow/c/ |
| c_api_function_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| tf2xla_util.cc | 197 Graph* func_graph = fbody->graph; local
199 ReplaceArgUsageWithConstNode(func_graph, const_input_index_to_node));
206 GraphToFunctionDef(*func_graph, new_func_name, &replace_fdef));
215 TF_RETURN_IF_ERROR(CopyAssociatedFunctions(func_graph, lookup_fld, fld));
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| trt_convert.py | 31 from tensorflow.python.framework import func_graph
296 converted_graph = func_graph.FuncGraph(func.graph.name)
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 385 FunctionBody* func_graph = nullptr; member in struct:tensorflow::FunctionLibraryRuntimeImpl::Item
391 delete this->func_graph;
529 return iter->second->func_graph;
736 item->func_graph = fbody;
864 fbody = (*item)->func_graph;
[all...] |
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| mirrored_strategy_multigpu_test.py | 45 from tensorflow.python.framework import func_graph
277 with func_graph.FuncGraph("fg").as_default(), distribution.scope():
330 with func_graph.FuncGraph("fg").as_default(), distribution.scope():
[all...] |
| /external/tensorflow/tensorflow/python/keras/engine/ |
| base_layer.py | 33 from tensorflow.python.framework import func_graph
466 graph = func_graph.FuncGraph('graph')
[all...] |
| network.py | 32 from tensorflow.python.framework import func_graph
799 graph = func_graph.FuncGraph('build_graph')
[all...] |
| /external/tensorflow/tensorflow/python/keras/ |
| backend.py | 43 from tensorflow.python.framework import func_graph
460 _GRAPH = func_graph.FuncGraph('keras_graph')
492 graph = graph or func_graph.FuncGraph('keras_scratch_graph')
[all...] |
