Home
last modified time | relevance | path

Searched defs:cond_graph (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dwhile_v2.py424 def _build_while_op(loop_vars, cond_graph, body_graph, output_shapes, argument
608 def _create_grad_func(ys, xs, grads, cond_graph, body_graph, name, while_op, argument
1238 def _check_num_inputs_outputs(cond_graph, body_graph, num_flattened_loop_vars): argument
1262 def _build_cond_placeholders_name_prefix(cond_graph): argument
1266 def _duplicate_body_captures_in_cond(cond_graph, body_graph_captures): argument
Dcond_v2.py399 def _resolve_grad_inputs(cond_graph, grad_graph): argument
/external/tensorflow/tensorflow/c/
Dc_api.h916 TF_Graph* const cond_graph; member
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc405 std::unique_ptr<Graph> cond_graph; in FunctionalizeLoop() local
Dfunctionalize_control_flow_test.cc103 void ConditionalTestFixture::BuildCondGraph(Graph* cond_graph) { in BuildCondGraph()
220 Graph cond_graph(OpRegistry::Global()); in RunTest() local
/external/tensorflow/tensorflow/compiler/jit/
Dextract_outside_compilation_pass.cc1568 Graph* cond_graph = cond_fbody->graph; in RewriteHostWhileLoopCond() local