| /aosp_15_r20/external/pytorch/functorch/examples/ensembling/ |
| H A D | parallel_train.py | 62 def __init__(self, hidden_dim=32, n_classes=2): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_memory.py | 40 def __init__(self, hidden_dim, with_fsdp=False, with_checkpoint=False): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ |
| H A D | benchmark.py | 37 def __init__(self, input_dim, hidden_dim, output_dim, dtype): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_spawn_gloo.py | 189 def __init__(self, input_dim, hidden_dim, output_dim, hidden_layers): argument
|
| /aosp_15_r20/external/executorch/examples/models/llama/ |
| H A D | llama_transformer.py | 87 hidden_dim: Optional[int] = None variable in ModelArgs
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_compile.py | 439 def __init__(self, hidden_dim): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/ |
| H A D | common_dtensor.py | 143 def __init__(self, dim, hidden_dim, dropout_p): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchvision_models.py | 500 hidden_dim=256, argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_eager_transforms.py | 841 def __init__(self, hidden_dim=32, n_classes=2): argument
863 def __init__(self, hidden_dim=32, n_classes=2): argument
4378 def __init__(self, hidden_dim=32, n_classes=2): argument
4479 def __init__(self, hidden_dim=32, n_classes=2): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_repros.py | 1657 def fcnn(in_dim, out_dim, hidden_dim, activation=torch.nn.GELU): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 9687 def __init__(self, input_dim, hidden_dim): argument
|