Home
last modified time | relevance | path

Searched refs:backendToDeviceType (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DDeprecatedTypeProperties.h42 return backendToDeviceType(backend_); in device_type()
46 return backendToDeviceType(backend_) == kCUDA; in is_cuda()
/aosp_15_r20/external/pytorch/c10/core/
H A DBackend.h220 inline DeviceType backendToDeviceType(Backend b) { in backendToDeviceType() function
/aosp_15_r20/external/pytorch/torch/csrc/tensor/
H A Dpython_tensor.cpp460 return at::Device(c10::backendToDeviceType(default_backend)); in get_default_device()
/aosp_15_r20/external/pytorch/aten/src/ATen/templates/
H A DTensorBody.h237 …return to(options().device(backendToDeviceType(b)).layout(layout_from_backend(b)), /*non_blocking*… in toBackend()