Searched refs:func_graph_module (Results 1 – 6 of 6) sorted by relevance
19 from tensorflow.python.framework import func_graph as func_graph_module unknown70 structured_outputs = func_graph_module.convert_structure_to_signature(123 outer_graph = func_graph_module.FuncGraph(152 func_graph_module.func_graph_from_py_func(
23 from tensorflow.python.framework import func_graph as func_graph_module unknown81 func_graph_module.device_stack_has_callable(
46 from tensorflow.python.framework import func_graph as func_graph_module unknown621 backwards_graph = func_graph_module.FuncGraph(623 func_graph_module.func_graph_from_py_func(819 backwards_graph = func_graph_module.FuncGraph(901 forward_wrapper_graph = func_graph_module.FuncGraph(994 wrapped_backwards_graph = func_graph_module.FuncGraph(2453 self._captures_container = func_graph_module.CapturesContainer()2616 func_graph_module.func_graph_from_py_func(3254 if func_graph_module is None or memory is None or self._func_graph is None:3257 func_graph_module.dismantle_func_graph(self._func_graph)
80 from tensorflow.python.framework import func_graph as func_graph_module unknown520 func_graph_module.dismantle_func_graph(self.func_graph)750 lifted_initializer_graph = func_graph_module.FuncGraph("initializer")1152 isinstance(arg, func_graph_module.UnknownArgument)
31 from tensorflow.python.framework import func_graph as func_graph_module unknown80 true_graph = func_graph_module.func_graph_from_py_func(87 false_graph = func_graph_module.func_graph_from_py_func(430 return func_graph_module.func_graph_from_py_func(614 t for t in func_graph_module.flatten(grad_graph.structured_outputs)666 t for t in func_graph_module.flatten(branch_outs) if t is not None760 return func_graph_module.pack_sequence_as(structured_outputs,1046 func_graph_module.func_graph_from_py_func(
30 from tensorflow.python.framework import func_graph as func_graph_module unknown159 cond_graph = func_graph_module.func_graph_from_py_func(222 body_graph = func_graph_module.func_graph_from_py_func(447 cond_grad_graph = func_graph_module.func_graph_from_py_func(695 grad_func_graph = func_graph_module.func_graph_from_py_func(874 assert isinstance(tensor.graph, func_graph_module.FuncGraph)1374 if isinstance(graph, func_graph_module.FuncGraph):