Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dtensor_impl.cpp23 struct LTCGuardImpl : public c10::impl::DeviceGuardImplInterface { struct
24 at::DeviceType type() const override { in type()
28 c10::Device exchangeDevice(c10::Device device) const override { in exchangeDevice()
35 c10::Device getDevice() const override { in getDevice()
39 void setDevice(c10::Device device) const override { in setDevice()
44 void uncheckedSetDevice(c10::Device device) const noexcept override { in uncheckedSetDevice()
49 c10::Stream getStream(c10::Device device) const noexcept override { in getStream()
54 c10::Stream exchangeStream(c10::Stream _unused) const noexcept override { in exchangeStream()
58 c10::DeviceIndex deviceCount() const noexcept override { in deviceCount()