Searched refs:_set_func_attr (Results 1 – 4 of 4) sorted by relevance
140 if_op._set_func_attr("then_branch", util.create_new_tf_function(true_graph))141 if_op._set_func_attr("else_branch",
288 while_op._set_func_attr("cond", util.create_new_tf_function(cond_graph))289 while_op._set_func_attr("body", util.create_new_tf_function(body_graph))
644 op._set_func_attr("f", self._forward_function.name)
2535 def _set_func_attr(self, attr_name, func_name): member in Operation