| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | transformercoder.h | 26 TransformerEncoderLayer encoder_layer,
35 TORCH_ARG(TransformerEncoderLayer, encoder_layer) = nullptr;
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fully_shard/ |
| H A D | test_fully_shard_model_checkpoint.py | 16 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
140 policy=ModuleWrapPolicy({TransformerEncoderLayer, TransformerDecoderLayer}),
163 policy=ModuleWrapPolicy({TransformerDecoderLayer, TransformerEncoderLayer}),
|
| H A D | test_fully_shard_compile.py | 69 nn.TransformerEncoderLayer,
87 module, (nn.TransformerEncoderLayer, nn.TransformerDecoderLayer)
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_wrap.py | 34 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
453 transformer_layer_cls={TransformerEncoderLayer, TransformerDecoderLayer},
461 {TransformerEncoderLayer, TransformerDecoderLayer}
469 {TransformerEncoderLayer, TransformerDecoderLayer}
521 module, (TransformerEncoderLayer, TransformerDecoderLayer)
531 elif isinstance(module, TransformerEncoderLayer):
|
| H A D | test_fsdp_hybrid_shard.py | 26 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
377 {TransformerEncoderLayer, TransformerDecoderLayer},
405 {TransformerEncoderLayer, TransformerDecoderLayer},
|
| H A D | test_fsdp_clip_grad_norm.py | 17 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
143 TransformerEncoderLayer,
|
| H A D | test_fsdp_sharded_grad_scaler.py | 20 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
229 TransformerEncoderLayer,
|
| H A D | test_fsdp_mixed_precision.py | 26 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
836 TransformerEncoderLayer,
966 TransformerEncoderLayer,
1328 {nn.Transformer, nn.TransformerEncoderLayer, nn.TransformerDecoderLayer}
|
| H A D | test_fsdp_state_dict.py | 40 from torch.nn import Linear, Module, TransformerDecoderLayer, TransformerEncoderLayer
436 {TransformerEncoderLayer, TransformerDecoderLayer}
462 {TransformerEncoderLayer, TransformerDecoderLayer}
1201 {TransformerEncoderLayer, TransformerDecoderLayer}
|
| H A D | test_fsdp_backward_prefetch.py | 87 {nn.TransformerEncoderLayer, nn.TransformerDecoderLayer}
|
| H A D | test_fsdp_use_orig_params.py | 29 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
136 TransformerEncoderLayer,
243 TransformerEncoderLayer,
|
| H A D | test_fsdp_misc.py | 31 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
676 module_classes = {TransformerEncoderLayer, TransformerDecoderLayer}
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | transformer.cpp | 62 TransformerEncoderLayer model = in transformer_encoder_layer_test_helper()
63 get_a_test_layer<TransformerEncoderLayer, TransformerEncoderLayerOptions>( in transformer_encoder_layer_test_helper()
232 TEST_F(TransformerTest, TransformerEncoderLayer) { in TEST_F() argument
573 TransformerEncoderLayer encoder_layer = in transformer_encoder_test_helper()
574 get_a_test_layer<TransformerEncoderLayer, TransformerEncoderLayerOptions>( in transformer_encoder_test_helper()
764 c10::str(TransformerEncoderLayer(4, 2)), in TEST_F()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_transformers.py | 218 layer = nn.TransformerEncoderLayer(
305 model = torch.nn.TransformerEncoderLayer(
332 torch.nn.TransformerEncoderLayer(
419 …torch.nn.TransformerEncoderLayer(d_model=4, nhead=2, dim_feedforward=16, dropout=0.0, batch_first=…
465 layer = nn.TransformerEncoderLayer(
651 encoder_layer = nn.TransformerEncoderLayer(
722 layer = torch.nn.TransformerEncoderLayer(d_model, 1, 6, batch_first=True)
763 encoder_layer = nn.TransformerEncoderLayer(
784 encoder_layer = nn.TransformerEncoderLayer(
811 class MyCustomLayer(nn.TransformerEncoderLayer):
[all …]
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | transformer.py | 124 encoder_layer = TransformerEncoderLayer(
352 if not isinstance(encoder_layer, torch.nn.TransformerEncoderLayer):
619 class TransformerEncoderLayer(Module): class
|
| H A D | __init__.py | 164 TransformerEncoderLayer,
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | transformer.cpp | 19 TransformerEncoderLayer encoder_layer, in TransformerEncoderOptions()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | transformercoder.h | 40 TransformerEncoderLayer encoder_layer, in TransformerEncoderImpl()
|
| H A D | transformerlayer.h | 88 TORCH_MODULE(TransformerEncoderLayer);
|
| /aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/ |
| H A D | torchaudio_models.py | 434 from torch.nn import TransformerEncoder, TransformerEncoderLayer
442 encoder_layers = TransformerEncoderLayer(ninp, nhead, nhid, dropout)
|
| /aosp_15_r20/external/pytorch/test/package/ |
| H A D | test_dependency_api.py | 363 model = torch.nn.TransformerEncoderLayer(
|
| /aosp_15_r20/external/pytorch/test/distributed/_tensor/ |
| H A D | test_attention.py | 216 encoder_layer = nn.TransformerEncoderLayer(
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | quantization_ops.py | 79 nn.TransformerEncoderLayer(d_model=2, nhead=2), num_layers=1
|
| H A D | nn_ops.py | 256 nn.TransformerEncoderLayer(d_model=2, nhead=2), num_layers=1
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_models_onnxruntime.py | 401 encoder_layers = nn.TransformerEncoderLayer(ninp, nhead, nhid, dropout)
|