/external/tensorflow/tensorflow/python/framework/ |
D | function_def_to_graph.py | 34 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 …]
|
D | func_graph.py | 818 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 …]
|
D | function.py | 945 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 …]
|
D | auto_control_deps_utils.py | 36 def get_read_only_resource_input_indices_graph(func_graph): argument 42 for input_index, t in enumerate(func_graph.inputs):
|
D | auto_control_deps_test.py | 192 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()
|
D | function_def_to_graph_test.py | 213 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/ |
D | control_flow_util_v2.py | 28 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
|
D | cond_v2.py | 35 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 …]
|
D | control_flow_v2_func_graphs.py | 21 from tensorflow.python.framework import func_graph 25 class ControlFlowFuncGraph(func_graph.FuncGraph): 38 func_graph.override_func_graph_name_scope(
|
D | while_v2.py | 34 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 …]
|
D | while_v2_indexed_slices_rewriter.py | 23 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(
|
D | gradients_util.py | 36 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
|
D | op_selector_test.py | 21 from tensorflow.python.framework import func_graph 76 inner = func_graph.FuncGraph("inner")
|
D | script_ops.py | 34 from tensorflow.python.framework import func_graph 327 elif isinstance(graph, func_graph.FuncGraph):
|
/external/tensorflow/tensorflow/python/saved_model/ |
D | function_deserialization.py | 29 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():
|
D | function_serialization.py | 23 from tensorflow.python.framework import func_graph as func_graph_module 167 func_graph=outer_graph)
|
D | load_v1_in_v2.py | 29 from tensorflow.python.framework import func_graph 191 func_graph.convert_structure_to_signature(
|
D | BUILD | 441 "//tensorflow/python:func_graph", 536 "//tensorflow/python:func_graph", 552 "//tensorflow/python:func_graph",
|
/external/tensorflow/tensorflow/python/eager/ |
D | lift_to_graph_test.py | 25 from tensorflow.python.framework import func_graph 48 g = func_graph.FuncGraph('lifted') 75 g = func_graph.FuncGraph('lifted')
|
D | wrap_function.py | 30 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(
|
D | def_function.py | 36 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)
|
D | lift_to_graph.py | 24 from tensorflow.python.framework import func_graph 299 if (isinstance(base_graph, func_graph.FuncGraph) and 300 isinstance(graph, func_graph.FuncGraph)):
|
D | function.py | 53 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/ |
D | c_api_function_test.cc | 1216 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/ |
D | tf2xla_util.cc | 192 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()
|