Home
last modified time | relevance | path

Searched defs:tpu_platform (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc74 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in RunTransferWithOrdinal() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h223 TpuPlatform& tpu_platform() { in tpu_platform() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_tpu_device.cc105 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform(); in CheckIfTPUInitialized() local
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc3957 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in GetDeviceTopology() local