Home
last modified time | relevance | path

Searched defs:ToyModel (Results 1 – 21 of 21) sorted by relevance

/external/libaom/test/
Dav1_external_partition_test.cc38 typedef struct ToyModel { struct
39 TestData *data;
40 aom_ext_part_config_t config;
41 aom_ext_part_funcs_t funcs;
42 int mi_row;
43 int mi_col;
44 int frame_width;
45 int frame_height;
46 BLOCK_SIZE block_size;
47 } ToyModel; typedef
/external/pytorch/test/
Dtest_compile_benchmark_util.py24 class ToyModel(torch.nn.Module): class
/external/pytorch/test/inductor/
Dtest_debug_trace.py218 class ToyModel(torch.nn.Module): class
Dtest_inductor_freezing.py392 class ToyModel(nn.Module): class
Dtest_cuda_repro.py1261 class ToyModel(torch.nn.Module): class
/external/pytorch/test/dynamo/
Dtest_compile.py14 class ToyModel(torch.nn.Module): class
Dtest_structured_trace.py349 class ToyModel(torch.nn.Module): class
Dtest_modules.py2375 class ToyModel(torch.nn.Module): class
2390 class ToyModel(torch.nn.Module): class
2424 class ToyModel(torch.nn.Module): class
2470 class ToyModel(torch.nn.Module): class
2493 class ToyModel(torch.nn.Module): class
2560 class ToyModel(torch.nn.Module): class
Dtest_logging.py215 class ToyModel(torch.nn.Module): class
Dtest_higher_order_ops.py2224 class ToyModel(torch.nn.Module): class
/external/executorch/exir/tests/
Dtest_memory_format_ops_pass.py102 class ToyModel(torch.nn.Module): class
/external/pytorch/benchmarks/dynamo/
Ddist_util.py61 class ToyModel(nn.Module): class
/external/pytorch/test/distributed/_composable/
Dtest_contract.py12 class ToyModel(nn.Module): class
Dtest_replicate.py260 class ToyModel(nn.Module): class
Dtest_checkpoint.py43 class ToyModel(nn.Module): class
/external/pytorch/test/distributed/_composable/fsdp/
Dtest_fully_shard_mixed_precision.py402 class ToyModel(nn.Module): class
/external/pytorch/test/distributed/
Dtest_dynamo_distributed.py64 class ToyModel(nn.Module): class
1493 class ToyModel(nn.Module): class
1547 class ToyModel(nn.Module): class
Dtest_c10d_nccl.py3466 class ToyModel(nn.Module): class in SparseCollective
/external/pytorch/torch/testing/_internal/distributed/
Ddistributed_test.py4415 class ToyModel(nn.Module): class
7740 class ToyModel(nn.Module): class
7794 class ToyModel(nn.Module): class
7875 class ToyModel(torch.nn.Module): class
8178 class ToyModel(nn.Module): class
8695 class ToyModel(nn.Module): class
9395 class ToyModel(nn.Module): class
/external/pytorch/test/distributed/fsdp/
Dtest_fsdp_mixed_precision.py1254 class ToyModel(nn.Module): class
/external/pytorch/test/nn/
Dtest_module_hooks.py41 class ToyModel(nn.Module): class