Home
last modified time | relevance | path

Searched refs:outer_graph (Results 1 – 25 of 28) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dfunc_graph.py244 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 Dauto_control_deps.py357 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 Dfunction_def_to_graph.py144 graph = graph.outer_graph
235 graph = graph.outer_graph
H A Dops.py5992 outer_graph = None
6005 outer_graph = get_default_graph()
6006 … outer_device_stack = outer_graph._device_function_stack # pylint: disable=protected-access
6007outer_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 Dfunction.py758 def outer_graph(self): member in _FuncGraph
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_v2_func_graphs.py26 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 Dcond_v2.py115 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 Dcontrol_flow_util_v2.py49 graph = graph.outer_graph
71 func.add_to_graph(func_graph.outer_graph)
307 graph = graph.outer_graph
H A Dwhile_v2.py503 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 Dgradients_util.py520 curr_graph = curr_graph.outer_graph
622 graph = src_graph.outer_graph
628 graph = graph.outer_graph
H A Dcontrol_flow_util.py73 graph = graph.outer_graph
H A Dscript_ops.py358 graph = graph.outer_graph
/aosp_15_r20/art/compiler/optimizing/
H A Dnodes.cc2707 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 Dinliner.h37 HInliner(HGraph* outer_graph,
50 : HOptimization(outer_graph, name, stats), in HOptimization() argument
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dfunction_serialization.py123 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 Dload.py713 outer_graph = self._concrete_functions[proto.concrete_function].graph
714 captured_tensor = outer_graph.get_tensor_by_name(proto.name)
H A Dsave.py674 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 Dtensor_name_test.py40 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 Dwrap_function.py186 outer_graph = new_variable.graph
190 global_collection = outer_graph.get_collection_ref(
193 outer_graph.add_to_collection(
H A Ddef_function.py327 outer_graph = ops.get_default_graph()
334 [initial_value], outer_graph,
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/utils/
H A Dcontrol_flow_util.py36 graph = graph.outer_graph
/aosp_15_r20/external/tensorflow/tensorflow/python/debug/lib/
H A Ddumping_callback.py218 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 Dtensorflow.__internal__.-func-graph.pbtxt43 name: "outer_graph"
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcross_device_utils.py305 graph = graph.outer_graph
/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu.py2177 graph = graph.outer_graph

12