Searched refs:_set_func_attr (Results 1 – 4 of 4) sorted by relevance
159 if_op._set_func_attr("then_branch", util.create_new_tf_function(true_graph))160 if_op._set_func_attr("else_branch",
364 while_op._set_func_attr("cond", util.create_new_tf_function(cond_graph))365 while_op._set_func_attr("body", util.create_new_tf_function(body_graph))
767 op._set_func_attr("f", forward_function.name)
2290 def _set_func_attr(self, attr_name, func_name): member in Operation