Searched defs:d_hid (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/pipelining/ | ||
| H A D | test_transformer.py | 8 d_hid = 16 variable |
| H A D | test_microbatch.py | 15 d_hid = 512 variable |
| H A D | test_backward.py | 16 d_hid = 512 variable |
| H A D | test_pipe.py | 15 d_hid = 512 variable |
| H A D | test_schedule_multiproc.py | 40 d_hid = 512 variable |
| H A D | test_stage.py | 31 d_hid = 512 variable |
| H A D | model_registry.py | 10 def __init__(self, d_hid): argument |