| /aosp_15_r20/external/pytorch/test/distributed/_composable/ |
| H A D | test_checkpoint.py | 186 class MyModel(torch.nn.Module): class
225 class MyModel(torch.nn.Module): class
264 def fwd_ctx(mod: MyModel):
274 def recomp_ctx(mod: MyModel):
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_misc.py | 64 class MyModel(nn.Module): class
260 class MyModel(nn.Module): class
712 class MyModel(nn.Module): class
890 class MyModel(nn.Module): class
|
| H A D | test_wrap.py | 172 class MyModel(nn.Module): class
334 class MyModel(nn.Module): class
413 class MyModel(nn.Module): class
|
| H A D | test_checkpoint_wrapper.py | 58 class MyModel(nn.Module): class
237 class MyModel(nn.Module): class
|
| H A D | test_fsdp_mixed_precision.py | 355 class MyModel(nn.Module): class
777 class MyModel(torch.nn.Module): class
|
| H A D | test_fsdp_hybrid_shard.py | 81 class MyModel(nn.Module): class
|
| H A D | test_fsdp_meta.py | 84 class MyModel(nn.Module): class
|
| /aosp_15_r20/external/executorch/exir/tests/ |
| H A D | test_serde.py | 93 class MyModel(torch.nn.Module): class
113 class MyModel(torch.nn.Module): class
|
| H A D | test_passes.py | 367 class MyModel(nn.Module): class
402 class MyModel(nn.Module): class
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 4986 class MyModel(torch.nn.Module): class
9036 class MyModel(nn.Module): class
9358 class MyModel(nn.Module): class
9441 class MyModel(nn.Module): class
9652 class MyModel(nn.Module): class
9701 class MyModel(nn.Module): class
9796 class MyModel(torch.nn.Module): class
9916 class MyModel(torch.nn.Module): class
10401 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/executorch/devtools/size_analysis_tool/ |
| H A D | size_analysis_tool_test.py | 27 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/test/testdata/ |
| H A D | gen_tf_readvariableop_model.py | 31 class MyModel(module.Module): class
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 413 class MyModel(torch.nn.Module): class
421 class MyModel(torch.nn.Module): class
429 class MyModel(torch.nn.Module): class
437 class MyModel(torch.nn.Module): class
446 class MyModel(torch.nn.Module): class
453 class MyModel(torch.nn.Module): class
473 class MyModel(torch.nn.Module): class
482 class MyModel(torch.nn.Module): class
496 class MyModel(torch.nn.Module): class
512 class MyModel(torch.nn.Module): class
[all …]
|
| H A D | test_onnx_opset.py | 356 class MyModel(torch.nn.Module): class
|
| H A D | test_operators.py | 879 class MyModel(torch.nn.Module): class
892 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 4790 class MyModel(nn.Module): class
4893 class MyModel(torch.nn.Module): class
5809 class MyModel(torch.nn.Module): class
5872 class MyModel(torch.nn.Module): class
5932 class MyModel(torch.nn.Module): class
5963 class MyModel(torch.nn.Module): class
5994 class MyModel(torch.nn.Module): class
8998 class MyModel(nn.Module): class
9033 class MyModel(nn.Module): class
9070 class MyModel(nn.Module): class
[all …]
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_api.py | 21 class MyModel(nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_bundled_inputs.py | 328 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_layout_optim.py | 301 class MyModel(torch.nn.Module): class
|
| H A D | test_codecache.py | 62 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/ |
| H A D | load_test.py | 1883 class MyModel(autotrackable.AutoTrackable): class
2315 class MyModel(module.Module): class
2336 class MyModel(module.Module): class
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_init.py | 551 class MyModel(nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 1033 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_aotdispatch.py | 3588 class MyModel(torch.nn.Module): class
3643 class MyModel(torch.nn.Module): class
3698 class MyModel(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_dynamo_distributed.py | 555 class MyModel(torch.nn.Module): class
|