| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 127 auto* stream = device->tensorflow_gpu_device_info()->stream; in MakeReductionTestCase()
176 auto* stream = device->tensorflow_gpu_device_info()->stream; in MakeGatherTestCase()
234 auto* stream = device->tensorflow_gpu_device_info()->stream; in MakeBroadcastTestCase()
264 auto* stream = device->tensorflow_gpu_device_info()->stream; in VerifyResults()
334 auto* info = device->tensorflow_gpu_device_info(); in RunMultiNodeAllReduceTest()
335 auto* stream = device->tensorflow_gpu_device_info()->stream; in RunMultiNodeAllReduceTest()
383 auto* info = device->tensorflow_gpu_device_info(); in RunMultiNodeBroadcastTest()
384 auto* stream = device->tensorflow_gpu_device_info()->stream; in RunMultiNodeBroadcastTest()
478 auto* info = device->tensorflow_gpu_device_info(); in TYPED_TEST()
479 auto* stream = device->tensorflow_gpu_device_info()->stream; in TYPED_TEST()
[all …]
|
| D | collective_communicator.cc | 79 auto* gpu_info = col_ctx->op_ctx->device()->tensorflow_gpu_device_info(); in Enqueue()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | nccl_ops.cc | 109 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
140 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
176 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
210 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
246 auto* gpu_info = c->device()->tensorflow_gpu_device_info(); in ComputeAsync()
|
| D | gpu_device_array.h | 91 context_->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute( in Finalize()
|
| D | debug_ops.h | 753 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
780 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
812 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
845 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
871 context->device()->tensorflow_gpu_device_info()->event_mgr->ThenExecute(
|
| D | collective_nccl_test.cc | 216 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
279 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
288 auto* dev_info = device_->tensorflow_gpu_device_info(); in PrepareDeviceContext()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 53 const GpuDeviceInfo* tensorflow_gpu_device_info() const override { in tensorflow_gpu_device_info() function
54 return underlying_device_->tensorflow_gpu_device_info(); in tensorflow_gpu_device_info()
|
| D | collective_rma_local.cc | 140 src_dev->tensorflow_gpu_device_info(); in MemCpyAsync()
146 src_dev->tensorflow_gpu_device_info(); in MemCpyAsync()
|
| D | permuter_test.cc | 287 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
362 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
384 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoPermute()
|
| D | hierarchical_tree_broadcaster_test.cc | 451 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
480 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
620 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 | 319 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
438 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
462 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoGather()
|
| D | ring_reducer_test.cc | 341 auto* dev_info = dev->tensorflow_gpu_device_info(); in RunTest()
464 auto* dev_info = device_->tensorflow_gpu_device_info(); in InitTensor()
494 auto* dev_info = device_->tensorflow_gpu_device_info(); in DoReduce()
|
| D | ring_gatherer.cc | 141 col_ctx_->device->tensorflow_gpu_device_info(); in RunAsyncParts()
|
| D | process_function_library_runtime_test.cc | 177 CHECK(gpu_device_->tensorflow_gpu_device_info() != nullptr); in GPUToCPU()
179 gpu_device_->tensorflow_gpu_device_info()->default_context; in GPUToCPU()
195 CHECK(gpu_device_->tensorflow_gpu_device_info() != nullptr); in CPUToGPU()
197 gpu_device_->tensorflow_gpu_device_info()->default_context; in CPUToGPU()
|
| D | ring_reducer.cc | 191 col_ctx_->device->tensorflow_gpu_device_info(); in RunAsyncParts()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 68 auto di = device->tensorflow_gpu_device_info(); in PrepareCopy()
348 auto* dev_info = gpu_device->tensorflow_gpu_device_info(); in Sync()
357 auto* dev_info = gpu_device->tensorflow_gpu_device_info(); in SyncAll()
376 auto* dev_info = device->tensorflow_gpu_device_info(); in MemoryDebugString()
|
| D | gpu_util_platform_specific.cc | 53 device->tensorflow_gpu_device_info(); in ThenExecute()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 89 device_->tensorflow_gpu_device_info()->default_context; in RunComposedOp()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 87 platform_id = device->tensorflow_gpu_device_info() in XlaPlatformInfoFromDevice()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 166 virtual const GpuDeviceInfo* tensorflow_gpu_device_info() const { in tensorflow_gpu_device_info() function
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | core.cc | 28 return d == nullptr || d->tensorflow_gpu_device_info() == nullptr; in IsCPU()
|
| D | tensor_handle.cc | 977 const bool dst_cpu = dstd->tensorflow_gpu_device_info() == nullptr; in CopyToDevice()
978 const bool src_cpu = srcd->tensorflow_gpu_device_info() == nullptr; in CopyToDevice()
1006 src_device_context = srcd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
1010 dst_device_context = dstd->tensorflow_gpu_device_info()->default_context; in CopyToDevice()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed.cc | 122 if (to_device->tensorflow_gpu_device_info()) { in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 528 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
537 << " gpu_info: " << src_dev->tensorflow_gpu_device_info(); in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | py_func.cc | 83 return d == nullptr || d->tensorflow_gpu_device_info() == nullptr; in IsCPUDevice()
|