Home
last modified time | relevance | path

Searched refs:torch_tensor_device_name (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/
H A Dquantization_utils.h12 inline std::string torch_tensor_device_name(const at::Tensor& ten) { in torch_tensor_device_name() function
28 torch_tensor_device_name(x))
34 torch_tensor_device_name(x))
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu498 inline std::string torch_tensor_device_name(const at::Tensor& ten) { in torch_tensor_device_name() function
502 inline std::string torch_tensor_device_name( in torch_tensor_device_name() function
505 return torch_tensor_device_name(ten.value()); in torch_tensor_device_name()
524 torch_tensor_device_name(x))