Home
last modified time | relevance | path

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

/aosp_15_r20/external/libaom/test/
H A 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
/aosp_15_r20/external/pytorch/test/
H A Dtest_compile_benchmark_util.py24 class ToyModel(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_debug_trace.py218 class ToyModel(torch.nn.Module): class
H A Dtest_inductor_freezing.py392 class ToyModel(nn.Module): class
H A Dtest_cuda_repro.py1261 class ToyModel(torch.nn.Module): class
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py64 class ToyModel(nn.Module): class
1493 class ToyModel(nn.Module): class
1547 class ToyModel(nn.Module): class
H A Dtest_c10d_nccl.py3466 class ToyModel(nn.Module): class in SparseCollective
/aosp_15_r20/external/pytorch/test/dynamo/
H A 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
H A Dtest_compile.py14 class ToyModel(torch.nn.Module): class
H A Dtest_structured_trace.py349 class ToyModel(torch.nn.Module): class
H A Dtest_logging.py215 class ToyModel(torch.nn.Module): class
H A Dtest_higher_order_ops.py2224 class ToyModel(torch.nn.Module): class
/aosp_15_r20/external/executorch/exir/tests/
H A Dtest_memory_format_ops_pass.py102 class ToyModel(torch.nn.Module): class
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Ddist_util.py61 class ToyModel(nn.Module): class
/aosp_15_r20/external/pytorch/test/distributed/_composable/
H A Dtest_contract.py12 class ToyModel(nn.Module): class
H A Dtest_replicate.py260 class ToyModel(nn.Module): class
H A Dtest_checkpoint.py43 class ToyModel(nn.Module): class
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A 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
/aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/
H A Dtest_fully_shard_mixed_precision.py402 class ToyModel(nn.Module): class
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_mixed_precision.py1254 class ToyModel(nn.Module): class
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py41 class ToyModel(nn.Module): class