Searched defs:_get_mem_eff_drop_mask (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/ |
| D | test_transformers.py | 2641 … def _get_mem_eff_drop_mask(batch_size, n_heads, q_len, kv_len, p, seed, offset, device=device): function
2751 … def _get_mem_eff_drop_mask(batch_size, n_heads, q_len, kv_len, p, seed, offset, device=device): function
2995 …def _get_mem_eff_drop_mask(batch_size, n_heads, q_len, kv_len, dropout_p, seed, offset, device=dev… function
|