Searched refs:VirtualDeviceContext (Results 1 – 1 of 1) sorted by relevance
23 class VirtualDeviceContext : public DeviceContext { class25 VirtualDeviceContext() = default;38 void VirtualDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()47 void VirtualDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()93 *out_context = new VirtualDeviceContext; in TryGetDeviceContext()