Searched defs:body_graph (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | while_v2.py | 468 def _build_while_op(loop_vars, cond_graph, body_graph, output_shapes, argument
655 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
766 def _resolve_grad_captures(body_graph, body_grad_graph, while_op): argument
1298 def _check_num_inputs_outputs(cond_graph, body_graph, num_flattened_loop_vars): argument
1313 def _check_inputs_outputs_types_match(body_graph, flattened_loop_vars): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | control_flow_deps_to_chains.cc | 68 auto* body_graph = flib_def->Find(body_name); in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api.h | 937 TF_Graph* const body_graph; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_while.cc | 409 std::unique_ptr<Graph> body_graph; in FunctionalizeLoop() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_xla_computations_pass_test.cc | 294 std::unique_ptr<Graph> body_graph = MakeBodyGraph(); in TEST() local
|
| H A D | extract_outside_compilation_pass.cc | 1654 Graph* body_graph = body_fbody->graph; in RewriteHostWhileLoopBody() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | tfrt_graph_execution_state_test.cc | 1121 std::unique_ptr<Graph> body_graph = MakeBodyGraph(); in TEST() local
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | test_export.py | 6370 def forward(self, input, true_graph, body_graph): argument
|