/aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
H A D | func_graph.py | 244 outer_graph = ops.get_default_graph() 245 self._weak_outer_graph = weakref.ref(outer_graph) 246 while outer_graph.building_function: 247 outer_graph = outer_graph.outer_graph 250 self._fallback_outer_graph = outer_graph 270 elif self.outer_graph is not None and isinstance(self.outer_graph, 272 self.capture_by_value = self.outer_graph.capture_by_value 282 graph = self.outer_graph 324 self = self.outer_graph 445 graph = graph.outer_graph [all …]
|
H A D | auto_control_deps.py | 357 outer_graph = getattr(self._graph, "outer_graph", None) 358 if outer_graph is not None: 359 self._graph._add_control_dependencies = outer_graph._add_control_dependencies
|
H A D | function_def_to_graph.py | 144 graph = graph.outer_graph 235 graph = graph.outer_graph
|
H A D | ops.py | 5992 outer_graph = None 6005 outer_graph = get_default_graph() 6006 … outer_device_stack = outer_graph._device_function_stack # pylint: disable=protected-access 6007 …outer_graph._device_function_stack = innermost_nonempty_device_stack # pylint: disable=protected-… 6026 if outer_graph is not None: 6027 outer_graph._device_function_stack = outer_device_stack # pylint: disable=protected-access 7349 return _get_enclosing_context(graph.outer_graph)
|
H A D | function.py | 758 def outer_graph(self): member in _FuncGraph
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
H A D | control_flow_v2_func_graphs.py | 26 outer_graph = self.outer_graph 31 …self._device_function_stack = outer_graph._device_function_stack.copy() # pylint: disable=protect… 35 self, self.outer_graph.get_name_scope())
|
H A D | cond_v2.py | 115 assert true_graph.outer_graph == if_op.graph 116 assert false_graph.outer_graph == if_op.graph 286 true_graph.outer_graph = ops.get_default_graph() 287 false_graph.outer_graph = ops.get_default_graph() 461 if t.graph != grad_graph.outer_graph: 482 assert t.graph == cond_graph.outer_graph 1076 assert branch_graph.outer_graph == case_op.graph 1226 bg.outer_graph = ops.get_default_graph()
|
H A D | control_flow_util_v2.py | 49 graph = graph.outer_graph 71 func.add_to_graph(func_graph.outer_graph) 307 graph = graph.outer_graph
|
H A D | while_v2.py | 503 cond_graph.outer_graph = ops.get_default_graph() 504 body_graph.outer_graph = ops.get_default_graph() 801 assert t.graph == body_graph.outer_graph 1205 with self._forward_graph.outer_graph.as_default():
|
H A D | gradients_util.py | 520 curr_graph = curr_graph.outer_graph 622 graph = src_graph.outer_graph 628 graph = graph.outer_graph
|
H A D | control_flow_util.py | 73 graph = graph.outer_graph
|
H A D | script_ops.py | 358 graph = graph.outer_graph
|
/aosp_15_r20/art/compiler/optimizing/ |
H A D | nodes.cc | 2707 HInstruction* HGraph::InlineInto(HGraph* outer_graph, HInvoke* invoke) { in InlineInto() argument 2721 outer_graph->GetAllocator(), invoke->GetEnvironment()); in InlineInto() 2728 outer_graph->SetHasBoundsChecks(true); in InlineInto() 2731 outer_graph->SetHasLoops(true); in InlineInto() 2734 outer_graph->SetHasIrreducibleLoops(true); in InlineInto() 2737 outer_graph->SetHasDirectCriticalNativeCall(true); in InlineInto() 2740 outer_graph->SetHasTryCatch(true); in InlineInto() 2743 outer_graph->SetHasMonitorOperations(true); in InlineInto() 2746 outer_graph->SetHasTraditionalSIMD(true); in InlineInto() 2749 outer_graph->SetHasPredicatedSIMD(true); in InlineInto() [all …]
|
H A D | inliner.h | 37 HInliner(HGraph* outer_graph, 50 : HOptimization(outer_graph, name, stats), in HOptimization() argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
H A D | function_serialization.py | 123 outer_graph = func_graph_module.FuncGraph( 131 with outer_graph.as_default(): 154 func_graph=outer_graph) 160 outer_graph, spec=concrete_function._function_spec)
|
H A D | load.py | 713 outer_graph = self._concrete_functions[proto.concrete_function].graph 714 captured_tensor = outer_graph.get_tensor_by_name(proto.name)
|
H A D | save.py | 674 outer_fn = func_graph_map.get(outer_fn.graph.outer_graph) 682 outer_fn = func_graph_map.get(outer_fn.graph.outer_graph)
|
/aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
H A D | tensor_name_test.py | 40 with tf.Graph().as_default() as outer_graph: 46 with tf.compat.v1.Session(graph=outer_graph) as sess:
|
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
H A D | wrap_function.py | 186 outer_graph = new_variable.graph 190 global_collection = outer_graph.get_collection_ref( 193 outer_graph.add_to_collection(
|
H A D | def_function.py | 327 outer_graph = ops.get_default_graph() 334 [initial_value], outer_graph,
|
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/ |
H A D | control_flow_util.py | 36 graph = graph.outer_graph
|
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/ |
H A D | dumping_callback.py | 218 if hasattr(graph, "outer_graph") and graph.outer_graph: 219 return self._get_context_id(graph.outer_graph)
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.__internal__.-func-graph.pbtxt | 43 name: "outer_graph"
|
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
H A D | cross_device_utils.py | 305 graph = graph.outer_graph
|
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
H A D | tpu.py | 2177 graph = graph.outer_graph
|