Home
last modified time | relevance | path

Searched defs:TENSOR_ON_CUDA_GPU (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/
H A Dquantization_utils.h30 #define TENSOR_ON_CUDA_GPU(x) \ macro
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu520 #define TENSOR_ON_CUDA_GPU(x) \ macro