Searched refs:all_model_fn_ops (Results 1 – 1 of 1) sorted by relevance
1631 all_model_fn_ops = []1635 all_model_fn_ops.append(1657 all_model_fn_ops.append(1669 return self._merge_train(all_model_fn_ops, train_op_fn)1671 return self._merge_infer(all_model_fn_ops)1673 return self._merge_eval(all_model_fn_ops)1695 def _merge_train(self, all_model_fn_ops, train_op_fn): argument1709 for m in all_model_fn_ops:1726 def _merge_infer(self, all_model_fn_ops): argument1737 for head, m in zip(self._heads, all_model_fn_ops):[all …]