Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/lazy/ts_backend/
Dts_backend_impl.cpp26 struct TSBackendDeviceType : public BackendDeviceType { struct
27 TSBackendDeviceType() = delete;
28 TSBackendDeviceType(c10::DeviceType deviceType) in TSBackendDeviceType() function
49 default_device_type_ = std::make_shared<TSBackendDeviceType>(type); in TSBackendImpl()
155 default_device_type_ = std::make_shared<TSBackendDeviceType>( in SetDefaultDeviceType()
185 std::shared_ptr<TSBackendDeviceType> default_device_type_;