| /aosp_15_r20/external/tensorflow/tensorflow/core/nccl/ |
| H A D | nccl_manager_test.cc | 127 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 …]
|
| H A D | collective_communicator.cc | 91 col_ctx->op_ctx->device()->tensorflow_accelerator_device_info(); in Enqueue()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | nccl_ops.cc | 109 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()
|
| H A D | debug_ops.h | 755 ->tensorflow_accelerator_device_info()
789 ->tensorflow_accelerator_device_info()
828 ->tensorflow_accelerator_device_info()
862 ->tensorflow_accelerator_device_info()
889 ->tensorflow_accelerator_device_info()
|
| H A D | collective_nccl_test.cc | 218 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()
|
| H A D | gpu_device_array.h | 92 ->tensorflow_accelerator_device_info() in Finalize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| H A D | tensor_util.cc | 58 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | light_outside_compilation_kernels_for_test.cc | 92 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_util.cc | 68 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()
|
| H A D | gpu_device_test.cc | 157 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()
|
| H A D | gpu_util_platform_specific.cc | 53 device->tensorflow_accelerator_device_info(); in ThenExecute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_util.cc | 63 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()
|
| H A D | pluggable_device_context.cc | 54 device->tensorflow_accelerator_device_info(); in ThenExecute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | renamed_device.h | 53 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()
|
| H A D | collective_rma_local.cc | 140 src_dev->tensorflow_accelerator_device_info(); in MemCpyAsync()
146 src_dev->tensorflow_accelerator_device_info(); in MemCpyAsync()
|
| H A D | collective_test_util.cc | 268 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()
|
| H A D | process_function_library_runtime_test.cc | 179 CHECK(gpu_device_->tensorflow_accelerator_device_info() != nullptr); in GPUToCPU()
181 gpu_device_->tensorflow_accelerator_device_info()->default_context; in GPUToCPU()
197 CHECK(gpu_device_->tensorflow_accelerator_device_info() != nullptr); in CPUToGPU()
199 gpu_device_->tensorflow_accelerator_device_info()->default_context; in CPUToGPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | get_compiler_ir.cc | 84 dev->tensorflow_accelerator_device_info()->stream == nullptr) { in GetCompilerIr()
140 dev->tensorflow_accelerator_device_info()) { in GetCompilerIr()
|
| H A D | xla_platform_info.cc | 126 platform_id = device->tensorflow_accelerator_device_info() in XlaPlatformInfoFromDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | collective_rma_distributed.cc | 142 if (to_device->tensorflow_accelerator_device_info()) { in RecvFromPeer()
194 if (to_device->tensorflow_accelerator_device_info()) { in RecvFromPeer()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | tensor_handle.cc | 977 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()
|
| H A D | core.cc | 28 return d == nullptr || d->tensorflow_accelerator_device_info() == nullptr; in IsCPU()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | unary_ops_composition_test.cc | 89 device_->tensorflow_accelerator_device_info()->default_context; in RunComposedOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_worker_service.cc | 533 if (src_dev->tensorflow_accelerator_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
542 << src_dev->tensorflow_accelerator_device_info(); in GrpcRecvTensorAsync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | device_base.h | 169 virtual const AcceleratorDeviceInfo* tensorflow_accelerator_device_info() in tensorflow_accelerator_device_info() function
|