Searched defs:D_in (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/test/ |
| D | test_throughput_benchmark.py | 9 def __init__(self, D_in, H, D_out): argument
24 def __init__(self, D_in, H, D_out): argument
|
| /external/pytorch/test/profiler/ |
| D | test_profiler.py | 820 def __init__(self, N, D_in, D_out): argument
832 def __init__(self, D_in, H, D_out): argument
|