Searched defs:eos_token (Results 1 – 2 of 2) sorted by relevance
1269 def _mask_probs(probs, eos_token, finished): argument
1538 def dense_to_sparse(tensor, eos_token=0, outputs_collections=None, scope=None): argument