| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 99 auto* stream = device->tensorflow_gpu_device_info()->stream; in MakeReductionTestCase()
148 auto* stream = device->tensorflow_gpu_device_info()->stream; in MakeGatherTestCase()
195 auto* stream = device->tensorflow_gpu_device_info()->stream; in VerifyResults()
270 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST()
271 auto* stream = device->tensorflow_gpu_device_info()->stream; in TYPED_TEST()
316 auto* stream = device->tensorflow_gpu_device_info()->stream; in TYPED_TEST()
338 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST()
339 auto* stream = device->tensorflow_gpu_device_info()->stream; in TYPED_TEST()
382 auto* event_mgr = device->tensorflow_gpu_device_info()->event_mgr; in TYPED_TEST()
383 auto* stream = device->tensorflow_gpu_device_info()->stream; in TYPED_TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 105 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
137 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
174 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
209 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
246 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
|
| D | collective_nccl_reducer_test.cc | 182 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
237 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
261 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoReduce()
|
| D | gpu_device_array.h | 89 context_->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in Finalize()
|
| D | collective_nccl_reducer.cc | 125 auto* gpu_info = col_ctx_->op_ctx->device()->tensorflow_gpu_device_info(); in Run()
|
| D | check_numerics_op.cc | 212 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsync()
|
| D | where_op.cc | 360 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in ComputeAsyncType()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 261 const bool dst_cpu = dstd->tensorflow_gpu_device_info() == nullptr; in CopyToDevice()
262 const bool src_cpu = srcd->tensorflow_gpu_device_info() == nullptr; in CopyToDevice()
286 src_device_context = srcd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
290 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
|
| D | kernel_and_device.cc | 225 if (device_->tensorflow_gpu_device_info() != nullptr) { in Run()
226 device_context = device_->tensorflow_gpu_device_info()->default_context; in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 54 const GpuDeviceInfo* tensorflow_gpu_device_info() const override { in tensorflow_gpu_device_info() function
55 return underlying_->tensorflow_gpu_device_info(); in tensorflow_gpu_device_info()
|
| D | collective_rma_local.cc | 105 src_dev->tensorflow_gpu_device_info(); in MemCpyAsync()
111 src_dev->tensorflow_gpu_device_info(); in MemCpyAsync()
|
| D | process_function_library_runtime_test.cc | 108 CHECK(gpu_device_->tensorflow_gpu_device_info() != nullptr); in GPUToCPU()
110 gpu_device_->tensorflow_gpu_device_info()->default_context; in GPUToCPU()
132 CHECK(gpu_device_->tensorflow_gpu_device_info() != nullptr); in CPUToGPU()
134 gpu_device_->tensorflow_gpu_device_info()->default_context; in CPUToGPU()
|
| D | hierarchical_tree_broadcaster_test.cc | 437 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
471 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
616 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
642 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoBroadcast()
|
| D | ring_gatherer_test.cc | 308 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
427 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
454 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoGather()
|
| D | ring_reducer_test.cc | 328 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
451 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
483 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoReduce()
|
| D | ring_gatherer.cc | 136 col_ctx_->device->tensorflow_gpu_device_info(); in RunAsyncParts()
|
| D | ring_reducer.cc | 172 col_ctx_->device->tensorflow_gpu_device_info(); in RunAsyncParts()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 70 auto di = device->tensorflow_gpu_device_info(); in PrepareCopy()
347 auto* dev_info = gpu_device->tensorflow_gpu_device_info(); in Sync()
356 auto* dev_info = gpu_device->tensorflow_gpu_device_info(); in SyncAll()
375 auto* dev_info = device->tensorflow_gpu_device_info(); in MemoryDebugString()
|
| D | gpu_util_platform_specific.cc | 51 device->tensorflow_gpu_device_info(); in ThenExecute()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 116 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
125 << " gpu_info: " << src_dev->tensorflow_gpu_device_info(); in GrpcRecvTensorAsync()
|
| D | gdr_memory_manager.cc | 406 if (copy && device->tensorflow_gpu_device_info() && !on_host) { in TransportOptionsFromTensor()
492 if (copy && device->tensorflow_gpu_device_info() && !on_host) { in TensorFromTransportOptions()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 163 virtual const GpuDeviceInfo* tensorflow_gpu_device_info() const { in tensorflow_gpu_device_info() function
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 118 if (to_device->tensorflow_gpu_device_info()) { in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 494 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
503 << " gpu_info: " << src_dev->tensorflow_gpu_device_info(); in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 241 if (src_dev->tensorflow_gpu_device_info() && in AddRequest()
|