Searched defs:fn_cuda (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor.py | 186 def define_custom_op_for_test(id_, fn_cpu, fn_cuda, fn_xpu, fn_meta, tags=()): argument
198 def define_custom_op_2_for_test(id_, fn_cpu, fn_cuda, fn_xpu, fn_meta, tags=()): argument
212 def define_custom_op_3_for_test(id_, fn_cpu, fn_cuda, fn_xpu, fn_meta, tags=()): argument
|