Searched refs:_merge_train (Results 1 – 1 of 1) sorted by relevance
1669 return self._merge_train(all_model_fn_ops, train_op_fn)1695 def _merge_train(self, all_model_fn_ops, train_op_fn): member in _MultiHead