Home
last modified time | relevance | path

Searched refs:create_new_tf_function (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcond_v2.py162 if_op._set_func_attr("then_branch", util.create_new_tf_function(true_graph))
164 util.create_new_tf_function(false_graph))
273 util.create_new_tf_function(true_graph),
274 util.create_new_tf_function(false_graph),
1125 util.create_new_tf_function(branch_graph)
1212 [util.create_new_tf_function(g) for g in branch_graphs],
H A Dwhile_v2.py412 while_op._set_func_attr("cond", util.create_new_tf_function(cond_graph))
413 while_op._set_func_attr("body", util.create_new_tf_function(body_graph))
486 util.create_new_tf_function(cond_graph),
487 util.create_new_tf_function(body_graph),
H A Dcontrol_flow_util_v2.py60 def create_new_tf_function(func_graph): function