Home
last modified time | relevance | path

Searched refs:tensorflow_accelerator_device_info (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc127 auto* stream = device->tensorflow_accelerator_device_info()->stream; in MakeReductionTestCase()
176 auto* stream = device->tensorflow_accelerator_device_info()->stream; in MakeGatherTestCase()
234 auto* stream = device->tensorflow_accelerator_device_info()->stream; in MakeBroadcastTestCase()
264 auto* stream = device->tensorflow_accelerator_device_info()->stream; in VerifyResults()
334 auto* info = device->tensorflow_accelerator_device_info(); in RunMultiNodeAllReduceTest()
335 auto* stream = device->tensorflow_accelerator_device_info()->stream; in RunMultiNodeAllReduceTest()
383 auto* info = device->tensorflow_accelerator_device_info(); in RunMultiNodeBroadcastTest()
384 auto* stream = device->tensorflow_accelerator_device_info()->stream; in RunMultiNodeBroadcastTest()
478 auto* info = device->tensorflow_accelerator_device_info(); in TYPED_TEST()
479 auto* stream = device->tensorflow_accelerator_device_info()->stream; in TYPED_TEST()
[all …]
Dcollective_communicator.cc91 col_ctx->op_ctx->device()->tensorflow_accelerator_device_info(); in Enqueue()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dtensor_util.cc58 const bool src_cpu = srcd->tensorflow_accelerator_device_info() == nullptr; in TransferTensorToDevice()
59 const bool dst_cpu = dstd->tensorflow_accelerator_device_info() == nullptr; in TransferTensorToDevice()
84 srcd->tensorflow_accelerator_device_info()->default_context; in TransferTensorToDevice()
89 dstd->tensorflow_accelerator_device_info()->default_context; in TransferTensorToDevice()
/external/tensorflow/tensorflow/core/kernels/
Dnccl_ops.cc109 auto* gpu_info = c->device()->tensorflow_accelerator_device_info(); in ComputeAsync()
140 auto* gpu_info = c->device()->tensorflow_accelerator_device_info(); in ComputeAsync()
176 auto* gpu_info = c->device()->tensorflow_accelerator_device_info(); in ComputeAsync()
210 auto* gpu_info = c->device()->tensorflow_accelerator_device_info(); in ComputeAsync()
246 auto* gpu_info = c->device()->tensorflow_accelerator_device_info(); in ComputeAsync()
Dgpu_device_array.h92 ->tensorflow_accelerator_device_info() in Finalize()
Ddebug_ops.h755 ->tensorflow_accelerator_device_info()
789 ->tensorflow_accelerator_device_info()
828 ->tensorflow_accelerator_device_info()
862 ->tensorflow_accelerator_device_info()
889 ->tensorflow_accelerator_device_info()
Dcollective_nccl_test.cc218 auto* dev_info = dev->tensorflow_accelerator_device_info(); in RunTest()
281 auto* dev_info = device_->tensorflow_accelerator_device_info(); in InitTensor()
290 auto* dev_info = device_->tensorflow_accelerator_device_info(); in PrepareDeviceContext()
Dsparse_to_dense_op_gpu.cu.cc228 c->device()->tensorflow_accelerator_device_info()->event_mgr->ThenExecute( in operator ()()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dlight_outside_compilation_kernels_for_test.cc92 ctx->device()->tensorflow_accelerator_device_info()->stream; in Compute()
145 ctx->device()->tensorflow_accelerator_device_info()->stream; in Compute()
210 ctx->device()->tensorflow_accelerator_device_info()->stream; in Compute()
270 ctx->device()->tensorflow_accelerator_device_info()->stream; in Compute()
336 ctx->device()->tensorflow_accelerator_device_info()->stream; in Compute()
/external/tensorflow/tensorflow/core/common_runtime/
Drenamed_device.h53 const DeviceBase::AcceleratorDeviceInfo* tensorflow_accelerator_device_info() in tensorflow_accelerator_device_info() function
55 return underlying_device_->tensorflow_accelerator_device_info(); in tensorflow_accelerator_device_info()
Dcollective_rma_local.cc140 src_dev->tensorflow_accelerator_device_info(); in MemCpyAsync()
146 src_dev->tensorflow_accelerator_device_info(); in MemCpyAsync()
Dcollective_test_util.cc268 auto* dev_info = device->tensorflow_accelerator_device_info(); in CopyTensorToDevice()
282 auto* dev_info = device->tensorflow_accelerator_device_info(); in CopyTensorToHost()
328 auto* dev_info = device->tensorflow_accelerator_device_info(); in RunCollective()
Dring_gatherer.cc141 col_ctx_->device->tensorflow_accelerator_device_info(); in RunAsyncParts()
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_util.cc63 auto di = device->tensorflow_accelerator_device_info(); in PrepareCopy()
271 auto* dev_info = device->tensorflow_accelerator_device_info(); in Sync()
280 auto* dev_info = device->tensorflow_accelerator_device_info(); in SyncAll()
Dpluggable_device_context.cc54 device->tensorflow_accelerator_device_info(); in ThenExecute()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc68 auto di = device->tensorflow_accelerator_device_info(); in PrepareCopy()
394 auto* dev_info = gpu_device->tensorflow_accelerator_device_info(); in Sync()
403 auto* dev_info = gpu_device->tensorflow_accelerator_device_info(); in SyncAll()
422 auto* dev_info = device->tensorflow_accelerator_device_info(); in MemoryDebugString()
Dgpu_util_platform_specific.cc53 device->tensorflow_accelerator_device_info(); in ThenExecute()
Dgpu_device_test.cc157 auto* device_info = device->tensorflow_accelerator_device_info(); in TEST_F()
186 auto* device_info = device->tensorflow_accelerator_device_info(); in TEST_F()
525 auto* device_info = device->tensorflow_accelerator_device_info(); in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dget_compiler_ir.cc84 dev->tensorflow_accelerator_device_info()->stream == nullptr) { in GetCompilerIr()
140 dev->tensorflow_accelerator_device_info()) { in GetCompilerIr()
Dxla_platform_info.cc126 platform_id = device->tensorflow_accelerator_device_info() in XlaPlatformInfoFromDevice()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcollective_rma_distributed.cc142 if (to_device->tensorflow_accelerator_device_info()) { in RecvFromPeer()
194 if (to_device->tensorflow_accelerator_device_info()) { in RecvFromPeer()
/external/tensorflow/tensorflow/compiler/tests/
Dunary_ops_composition_test.cc89 device_->tensorflow_accelerator_device_info()->default_context; in RunComposedOp()
/external/tensorflow/tensorflow/core/framework/
Ddevice_base.h169 virtual const AcceleratorDeviceInfo* tensorflow_accelerator_device_info() in tensorflow_accelerator_device_info() function
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcore.cc28 return d == nullptr || d->tensorflow_accelerator_device_info() == nullptr; in IsCPU()
Dtensor_handle.cc977 const bool dst_cpu = dstd->tensorflow_accelerator_device_info() == nullptr; in CopyToDevice()
978 const bool src_cpu = srcd->tensorflow_accelerator_device_info() == nullptr; in CopyToDevice()
1007 srcd->tensorflow_accelerator_device_info()->default_context; in CopyToDevice()
1012 dstd->tensorflow_accelerator_device_info()->default_context; in CopyToDevice()

123