Searched refs:src_device_context (Results 1 – 1 of 1) sorted by relevance
284 tensorflow::DeviceContext* src_device_context = nullptr; in CopyToDevice() local286 src_device_context = srcd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()302 tensorflow::CopyTensor::ViaDMA("copy", src_device_context, dst_device_context, in CopyToDevice()