| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/trainer/ |
| H A D | trainer.py | 33 def record_start(self, type, key, name, cuda=True): argument
53 def record_batch_start(self, key, cuda=True): argument
76 def record_forward_start(self, key, cuda=True): argument
99 def record_backward_start(self, key, cuda=True): argument
123 def methodmetric(name, type="method_metric", cuda=True): argument
|
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/server/ |
| H A D | server.py | 53 def record_start(self, type, key, name, cuda=True): argument
73 def record_straggler_start(self, key, cuda=True): argument
99 def record_batch_start(self, key, cuda=True): argument
124 def record_method(name, type="method_metric", cuda=True): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/algorithms/quantization/ |
| H A D | test_quantization.py | 213 cuda=False, argument
244 cuda=False, argument
280 cuda=False, argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 2008 cuda=False, argument
2142 cuda=False, argument
2397 cuda=False, argument
2528 self, tensor_out, tensor_in, group_id, rank, cuda=True, rank_to_GPU=None argument
2642 cuda=False, argument
3037 cuda=False, argument
3202 self, group, group_id, rank, cuda=False, rank_to_GPU=None, dtype=torch.float argument
3325 self, group, group_id, rank, cuda=False, rank_to_GPU=None argument
3423 self, group, group_id, rank, cuda=False, rank_to_GPU=None, dtype=torch.float argument
3549 self, tensor_out, tensor_in, group_id, rank, cuda=True, rank_to_GPU=None argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cuda.py | 198 self, name: str, kernel: str, metadata: Optional[str] = None, cuda=True argument
351 cuda=True, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/ |
| H A D | CUDAHooks.cpp | 44 namespace c10::cuda::_internal { namespace
48 namespace at::cuda::detail { namespace
|
| H A D | UnpackRaw.cuh | 5 namespace at::cuda::philox { namespace
|
| H A D | IndexUtils.cuh | 7 namespace at::cuda::detail { namespace
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | rnn.cpp | 11 bool test_RNN_xor(Func&& model_maker, bool cuda = false) { in test_RNN_xor()
444 void BidirectionalGRUReverseForward(bool cuda) { in BidirectionalGRUReverseForward()
498 void BidirectionalLSTMReverseForwardTest(bool cuda) { in BidirectionalLSTMReverseForwardTest()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_platform_id.cc | 19 namespace cuda { namespace
|
| /aosp_15_r20/external/pytorch/c10/cuda/impl/ |
| H A D | CUDAGuardImpl.cpp | 3 namespace c10::cuda::impl { namespace
|
| H A D | CUDATest.cpp | 8 namespace c10::cuda::impl { namespace
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | llvm_basic.cpp | 15 namespace at::cuda { namespace
|
| H A D | llvm_complex.cpp | 17 namespace at::cuda { namespace
|
| H A D | CUDAGraphsUtils.cuh | 15 namespace at::cuda { namespace
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | reduction_template.cuh | 2 namespace cuda { namespace
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/nvrtc_stub/ |
| H A D | ATenNVRTC.cpp | 4 namespace at { namespace cuda { namespace
|
| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAMiscFunctions.cpp | 4 namespace c10::cuda { namespace
|
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/shared/ |
| H A D | nvtx.cpp | 11 namespace torch::cuda::shared { namespace
|
| H A D | cusparselt.cpp | 14 namespace torch::cuda::shared { namespace
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/ |
| H A D | StreamTimer.cpp | 16 namespace at::cuda::tunable { namespace
|
| /aosp_15_r20/external/pytorch/torch/utils/benchmark/op_fuzzers/ |
| H A D | binary.py | 17 def __init__(self, seed, dtype=torch.float32, cuda=False): argument
|
| H A D | sparse_binary.py | 17 def __init__(self, seed, dtype=torch.float32, cuda=False): argument
|
| H A D | unary.py | 17 def __init__(self, seed, dtype=torch.float32, cuda=False): argument
|
| H A D | sparse_unary.py | 16 def __init__(self, seed, dtype=torch.float32, cuda=False): argument
|