Searched defs:in_feat (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/distributed/ |
| D | test_dynamo_distributed.py | 65 def __init__(self, in_feat=10, hidden_feat=5000, out_feat=5, ctx_manager=None): argument
84 device, bsz=20, in_feat=10, hidden_feat=5000, out_feat=5, ctx_manager=None argument
99 def __init__(self, in_feat=10, hidden_feat=5000, out_feat=5, ctx_manager=None): argument
116 device, bsz=20, in_feat=10, hidden_feat=5000, out_feat=5, ctx_manager=None argument
1080 self, bsz=20, in_feat=10, hidden_feat=5000, out_feat=5, ctx_manager=None argument
1494 def __init__(self, in_feat=10, hidden_feat=5000, out_feat=5): argument
1548 def __init__(self, in_feat=10, hidden_feat=5000, out_feat=5): argument
|