Home
last modified time | relevance | path

Searched refs:FuncGraph (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util_v2.py26 from tensorflow.python.framework.func_graph import FuncGraph
30 class CondBranchFuncGraph(FuncGraph):
38 class WhileCondFuncGraph(FuncGraph):
46 class WhileBodyFuncGraph(FuncGraph):
62 return isinstance(graph, FuncGraph)
Dgradients_util.py36 from tensorflow.python.framework.func_graph import FuncGraph
450 return (isinstance(graph, FuncGraph) or
455 if isinstance(func_graph, FuncGraph):
574 if isinstance(curr_graph, FuncGraph):
Dwhile_v2.py622 assert isinstance(tensor.graph, func_graph_module.FuncGraph)
942 while isinstance(outer_graph, func_graph_module.FuncGraph):
949 if isinstance(graph, func_graph_module.FuncGraph):
/external/tensorflow/tensorflow/python/framework/
Dfunc_graph.py126 class FuncGraph(ops.Graph): class
176 super(FuncGraph, self).__init__()
192 self.outer_graph, FuncGraph):
255 return super(FuncGraph, self).control_dependencies(control_inputs)
271 return super(FuncGraph, self).control_dependencies(filtered_control_inputs)
274 outer_cm = super(FuncGraph, self).as_default()
450 return super(FuncGraph, self).create_op(
478 while inner_graph is not None and isinstance(inner_graph, FuncGraph):
586 func_graph = FuncGraph(name, collections=collections,
588 assert isinstance(func_graph, FuncGraph)
Dfunction_def_to_graph.py28 from tensorflow.python.framework.func_graph import FuncGraph
50 func_graph = FuncGraph(fdef.signature.name)
Dconvert_to_constants.py91 output_graph = func_graph.FuncGraph(input_func.graph.name)
/external/tensorflow/tensorflow/python/eager/
Dlift_to_graph_test.py44 g = func_graph.FuncGraph('lifted')
Dlift_to_graph.py280 if (isinstance(base_graph, func_graph.FuncGraph) and
281 isinstance(graph, func_graph.FuncGraph)):
Dwrap_function.py176 pruned_graph = func_graph.FuncGraph(name)
263 self.graph = func_graph.FuncGraph(name, collections=collections, **kwargs)
Ddef_function.py345 lifted_initializer_graph = func_graph_module.FuncGraph("initializer")
Dfunction.py765 backwards_graph = func_graph_module.FuncGraph(
/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py78 with func_graph_lib.FuncGraph(name="guess_conversion").as_default():
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py296 converted_graph = func_graph.FuncGraph(func.graph.name)
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmirrored_strategy_multigpu_test.py277 with func_graph.FuncGraph("fg").as_default(), distribution.scope():
330 with func_graph.FuncGraph("fg").as_default(), distribution.scope():
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer.py466 graph = func_graph.FuncGraph('graph')
Dnetwork.py799 graph = func_graph.FuncGraph('build_graph')
/external/tensorflow/tensorflow/python/keras/
Dbackend.py460 _GRAPH = func_graph.FuncGraph('keras_graph')
492 graph = graph or func_graph.FuncGraph('keras_scratch_graph')