Searched defs:cond_graph (Results 1 – 7 of 7) 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
1298 def _check_num_inputs_outputs(cond_graph, body_graph, num_flattened_loop_vars): argument
1323 def _build_cond_placeholders_name_prefix(cond_graph): argument
1327 def _duplicate_body_captures_in_cond(cond_graph, body_graph_captures): argument
|
| H A D | cond_v2.py | 436 def _resolve_grad_inputs(cond_graph, grad_graph): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | control_flow_deps_to_chains.cc | 257 auto* cond_graph = flib_def->Find(cond_name); in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api.h | 931 TF_Graph* const cond_graph; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_while.cc | 404 std::unique_ptr<Graph> cond_graph; in FunctionalizeLoop() local
|
| H A D | functionalize_control_flow_test.cc | 106 void ConditionalTestFixture::BuildCondGraph(Graph* cond_graph) { in BuildCondGraph()
239 Graph cond_graph(OpRegistry::Global()); in RunTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | extract_outside_compilation_pass.cc | 1579 Graph* cond_graph = cond_fbody->graph; in RewriteHostWhileLoopCond() local
|