Home
last modified time | relevance | path

Searched defs:CUDA (Results 1 – 25 of 30) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/csrc/profiler/orchestration/
H A Dobserver.h18 CUDA, // CUDA kernels, runtime enumerator
33 CUDA, // CPU + CUDA events enumerator
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dauto_mixed_precision.h29 enum class AutoMixedPrecisionMode { CUDA, BF16, CPU }; enumerator
/aosp_15_r20/prebuilts/cmake/linux-x86/share/cmake-3.22/Help/release/
D3.8.rst37 CUDA subsection
/aosp_15_r20/external/pytorch/c10/core/
H A DDeviceType.h38 CUDA = 1, // CUDA. enumerator
H A DBackend.h31 CUDA, enumerator
/aosp_15_r20/external/pytorch/docs/source/community/
H A Dpersons_of_interest.rst204 CUDA section in Module-level maintainers
/aosp_15_r20/external/pytorch/functorch/benchmarks/
H A Doperator_authoring.py12 CUDA = False variable
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/
H A Dquantization_gpu.cu156 TORCH_LIBRARY_IMPL(quantization, CUDA, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DOps.cpp82 IMPL_SEND(CUDA) in IMPL_SEND() argument
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/NVPTX/
H A DNVPTX.h54 CUDA enumerator
/aosp_15_r20/external/llvm/lib/Target/NVPTX/
H A DNVPTX.h66 CUDA enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/clang/Basic/
DLangStandard.h41 CUDA, enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/clang/Basic/
DLangStandard.h40 CUDA, enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/clang/Basic/
DLangStandard.h40 CUDA, enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/clang/Basic/
DLangStandard.h40 CUDA, enumerator
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/NVPTX/
H A DNVPTX.h73 CUDA enumerator
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.h428 static inline DeprecatedTypeProperties& CUDA(ScalarType s) { in CUDA() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DEmbeddingBag.cu565 TORCH_LIBRARY_IMPL(quantized, CUDA, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/llvm/include/llvm/ADT/
H A DTriple.h163 CUDA, // NVIDIA CUDA enumerator
/aosp_15_r20/external/swiftshader/third_party/llvm-10.0/llvm/include/llvm/ADT/
H A DTriple.h181 CUDA, // NVIDIA CUDA enumerator
/aosp_15_r20/external/swiftshader/third_party/llvm-16.0/llvm/include/llvm/TargetParser/
H A DTriple.h209 CUDA, // NVIDIA CUDA enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r522817/include/llvm/TargetParser/
DTriple.h208 CUDA, // NVIDIA CUDA enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567/include/llvm/TargetParser/
DTriple.h209 CUDA, // NVIDIA CUDA enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r536225/include/llvm/TargetParser/
DTriple.h221 CUDA, // NVIDIA CUDA enumerator
/aosp_15_r20/prebuilts/clang/host/linux-x86/clang-r530567b/include/llvm/TargetParser/
DTriple.h209 CUDA, // NVIDIA CUDA enumerator

12