/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_padding.py | 64 The default vocab size for BertForMaskedLM is 30522. 193 from transformers import BertForMaskedLM 195 config_cls = BertForMaskedLM.config_class 204 model = BertForMaskedLM(config)
|
H A D | test_loop_ordering.py | 103 E.g. happens for BertForMaskedLM.
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
H A D | huggingface_models_list_cpu.txt | 6 BertForMaskedLM,16
|
H A D | huggingface_models_list.txt | 6 BertForMaskedLM,32
|
H A D | huggingface.yaml | 27 BertForMaskedLM: 2
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/ci_expected_accuracy/ |
H A D | cpu_aot_inductor_freezing_huggingface_inference.csv | 21 BertForMaskedLM,pass,0
|
H A D | aot_inductor_huggingface_inference.csv | 21 BertForMaskedLM,pass,0
|
H A D | dynamic_aot_eager_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|
H A D | cpu_inductor_freezing_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | dynamic_inductor_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|
H A D | inductor_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|
H A D | dynamo_eager_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|
H A D | dynamo_eager_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | inductor_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | aot_eager_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|
H A D | dynamic_inductor_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | cpu_inductor_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | cpu_inductor_amp_freezing_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | aot_eager_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | dynamic_cpu_inductor_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
H A D | dynamic_aot_eager_huggingface_inference.csv | 25 BertForMaskedLM,pass,0
|
/aosp_15_r20/external/pytorch/benchmarks/dynamo/ci_expected_accuracy/cu124/ |
H A D | inductor_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|
H A D | dynamic_inductor_huggingface_training.csv | 25 BertForMaskedLM,pass,5
|