Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/tpu/
Dtpu_executor_api.cc31 auto* tpu_platform = executor_api_fn->TpuPlatform_NewFn(); in IsStreamExecutorEnabled() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executor.h228 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.cc4465 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in GetDeviceTopology() local