| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | transformer.cpp | 253 TransformerDecoderLayer model = in transformer_decoder_layer_test_helper()
254 get_a_test_layer<TransformerDecoderLayer, TransformerDecoderLayerOptions>( in transformer_decoder_layer_test_helper()
438 TEST_F(TransformerTest, TransformerDecoderLayer) { in TEST_F() argument
459 TransformerDecoderLayer model = in transformer_decoder_layer_test_helper_gelu()
460 get_a_test_layer<TransformerDecoderLayer, TransformerDecoderLayerOptions>( in transformer_decoder_layer_test_helper_gelu()
819 c10::str(TransformerDecoderLayer(4, 2)), in TEST_F()
847 TransformerDecoderLayer decoder_layer = in transformer_decoder_test_helper()
848 get_a_test_layer<TransformerDecoderLayer, TransformerDecoderLayerOptions>( in transformer_decoder_test_helper()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | transformercoder.h | 57 TransformerDecoderLayer decoder_layer,
66 TORCH_ARG(TransformerDecoderLayer, decoder_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 | 70 nn.TransformerDecoderLayer,
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)
533 elif isinstance(module, TransformerDecoderLayer):
|
| 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
144 TransformerDecoderLayer,
|
| H A D | test_fsdp_sharded_grad_scaler.py | 20 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
230 TransformerDecoderLayer,
|
| H A D | test_fsdp_mixed_precision.py | 26 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
837 TransformerDecoderLayer,
967 TransformerDecoderLayer,
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
137 TransformerDecoderLayer,
244 TransformerDecoderLayer,
|
| H A D | test_fsdp_misc.py | 31 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
676 module_classes = {TransformerEncoderLayer, TransformerDecoderLayer}
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | transformer.cpp | 29 TransformerDecoderLayer decoder_layer, in TransformerDecoderOptions()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | transformercoder.h | 101 TransformerDecoderLayer decoder_layer, in TransformerDecoderImpl()
|
| H A D | transformerlayer.h | 192 TORCH_MODULE(TransformerDecoderLayer);
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | __init__.py | 162 TransformerDecoderLayer,
|
| H A D | transformer.py | 146 decoder_layer = TransformerDecoderLayer(
935 class TransformerDecoderLayer(Module): class
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | quantization_ops.py | 82 nn.TransformerDecoderLayer(d_model=2, nhead=2), num_layers=1
|
| H A D | nn_ops.py | 259 nn.TransformerDecoderLayer(d_model=2, nhead=2), num_layers=1
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 55 from torch.nn import TransformerDecoderLayer, TransformerEncoderLayer
345 TransformerDecoderLayer,
|
| /aosp_15_r20/external/pytorch/test/cpp_api_parity/ |
| H A D | parity-tracker.md | 100 torch::nn::TransformerDecoderLayer|Yes|No
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | nn.rst | 241 nn.TransformerDecoderLayer
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 1073 class TransformerDecoderLayer(nn.Module): class
1080 model = TransformerDecoderLayer(512, 8, batch_first)
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cuda_repro.py | 621 self.dec_layer = nn.TransformerDecoderLayer(
|