Searched defs:ctx_manager (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | logging_utils.py | 208 def ctx_manager(): function
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_checkpoint_wrapper.py | 114 def ctx_manager(): function
|