Searched refs:_set_func_attr (Results 1 – 4 of 4) sorted by relevance
165 if_op._set_func_attr("then_branch", util.create_new_tf_function(true_graph))166 if_op._set_func_attr("else_branch",
372 while_op._set_func_attr("cond", util.create_new_tf_function(cond_graph))373 while_op._set_func_attr("body", util.create_new_tf_function(body_graph))
749 op._set_func_attr("f", forward_function.name)
2500 def _set_func_attr(self, attr_name, func_name): member in Operation