Searched refs:SYCLDeviceContext (Results 1 – 3 of 3) sorted by relevance
28 class SYCLDeviceContext : public DeviceContext {30 SYCLDeviceContext() {} in SYCLDeviceContext() function32 ~SYCLDeviceContext() override {} in ~SYCLDeviceContext()
34 std::vector<SYCLDeviceContext*> m_sycl_context_; // ref counted119 m_sycl_context_.push_back(new SYCLDeviceContext()); in AddDevice()156 SYCLDeviceContext* GetSYCLContext(size_t i = 0) const {201 Allocator* cpu_allocator, SYCLDeviceContext* ctx) in SYCLDevice()227 SYCLDeviceContext* device_context_; // not owned
25 void SYCLDeviceContext::CopyCPUTensorToDevice(const Tensor *cpu_tensor, in CopyCPUTensorToDevice()102 void SYCLDeviceContext::CopyDeviceTensorToCPU(const Tensor *device_tensor, in CopyDeviceTensorToCPU()