Searched defs:mark_step (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/_lazy/ | ||
| D | __init__.py | 9 def mark_step(device: str = "", wait=False): function |
| /external/pytorch/test/lazy/ | ||
| D | test_ts_opinfo.py | 159 def foo(x, *, mark_step): argument |