Lines Matching refs:dev_info
67 const DeviceBase::GpuDeviceInfo** dev_info,
76 *dev_info = di;
122 const DeviceBase::GpuDeviceInfo* dev_info = nullptr;
124 Status s = PrepareCopy(dev, device_context, tensor, nullptr, &dev_info,
168 dev_info->event_mgr->ThenExecute(
197 const DeviceBase::GpuDeviceInfo* dev_info = nullptr;
199 Status s = PrepareCopy(src, send_dev_context, *input, output, &dev_info,
242 dev_info->event_mgr->ThenExecute(
264 const DeviceBase::GpuDeviceInfo* dev_info = nullptr;
267 &dev_info, &send_stream);
292 dev_info->event_mgr->ThenExecute(
309 const DeviceBase::GpuDeviceInfo* dev_info = nullptr;
312 &dev_info, &recv_stream);
338 dev_info->event_mgr->ThenExecute(
351 auto* dev_info = gpu_device->tensorflow_gpu_device_info();
352 if (!dev_info) {
355 return dev_info->stream->BlockHostUntilDone();
360 auto* dev_info = gpu_device->tensorflow_gpu_device_info();
361 if (!dev_info) {
364 if (!dev_info->stream->parent()->SynchronizeAllActivity() ||
365 !dev_info->stream->ok()) {
379 auto* dev_info = device->tensorflow_gpu_device_info();
380 if (!dev_info) {
387 auto s = dev_info->stream->parent()->SynchronousMemcpyD2H(
432 const DeviceBase::GpuDeviceInfo* dev_info = nullptr;
435 dst_gpu_tensor, &dev_info, &send_stream);
