Home
last modified time | relevance | path

Searched defs:original_device (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAGuard.h57 Device original_device() const { in original_device() function
119 std::optional<Device> original_device() const { in original_device() function
196 Device original_device() const { in original_device() function
/aosp_15_r20/external/pytorch/c10/core/
H A DDeviceGuard.h71 Device original_device() const { in original_device() function
171 std::optional<Device> original_device() const { in original_device() function
H A DStreamGuard.h78 Device original_device() const { in original_device() function
/aosp_15_r20/external/pytorch/c10/core/impl/
H A DInlineDeviceGuard.h185 Device original_device() const { in original_device() function
407 std::optional<Device> original_device() const { in original_device() function
H A DInlineStreamGuard.h115 Device original_device() const { in original_device() function
/aosp_15_r20/external/pytorch/aten/src/ATen/hip/impl/
H A DHIPGuardImplMasqueradingAsCUDA.h264 Device original_device() const { return guard_.original_device(); } in original_device() function
284 std::optional<Device> original_device() const { return guard_.original_device(); } in original_device() function
310 Device original_device() const { return guard_.original_device(); } in original_device() function
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSGuardImpl.h155 std::optional<Device> original_device() const { in original_device() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc1325 int original_device = 0; in CreateDevices() local