Searched defs:check_model_gpu (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/inductor/ | ||
| D | test_torchinductor_opinfo.py | 564 check_model_gpu = check_model_gpu variable in TestInductorOpInfo |
| D | test_compiled_optimizers.py | 600 check_model_gpu = check_model_gpu variable in CompiledOptimizerTests |
| D | test_torchinductor.py | 570 def check_model_gpu( function |