Searched defs:tpu_platform (Results 1 – 4 of 4) sorted by relevance
31 auto* tpu_platform = executor_api_fn->TpuPlatform_NewFn(); in IsStreamExecutorEnabled() local
228 TpuPlatform& tpu_platform() { in tpu_platform() function
105 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform(); in CheckIfTPUInitialized() local
4465 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in GetDeviceTopology() local