Searched defs:use_cuda (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_pg_wrapper.py | 72 def _test_collective_hang(self, wrapper_pg, use_cuda=False): argument
93 def _test_collectives_op_mismatch(self, wrapper_pg, use_cuda=False): argument
148 def _test_collective_shape_mismatch(self, wrapper_pg, use_cuda=False): argument
|
| /aosp_15_r20/external/pytorch/torch/utils/bottleneck/ |
| H A D | __main__.py | 95 def run_prof(use_cuda=False): argument
|
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_execution_trace.py | 50 def payload(self, use_cuda=False): argument
|
| H A D | test_profiler.py | 480 def payload(self, use_cuda=False): argument
|
| /aosp_15_r20/external/pytorch/torch/autograd/ |
| H A D | profiler_legacy.py | 40 use_cuda=False, argument
|
| H A D | profiler.py | 201 use_cuda=False, # Deprecated argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/ |
| H A D | compiler.cpp | 281 const bool use_cuda = device.is_cuda(); in compileKernel() local
|
| H A D | codegen.cpp | 368 const bool use_cuda) { in generateKernel()
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_convolution.py | 1235 use_cuda=False, argument
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_tracer.py | 916 def run_ge_tests(self, optimize, use_cuda): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 3083 def get_a_test_layer(use_cuda, activation, batch_first=False): argument
|