Searched defs:tpu_platform (Results 1 – 6 of 6) sorted by relevance
140 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform(); in InitializeInternal() local252 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform(); in Compute() local
190 tpu::TpuPlatformInterface* tpu_platform = in TPUCoreIDsToLocations() local224 tpu::TpuPlatformInterface* tpu_platform = in TPUCoreLocationsToIDs() local
31 auto* tpu_platform = executor_api_fn->TpuPlatform_NewFn(); in IsStreamExecutorEnabled() local
229 TpuPlatform& tpu_platform() { in tpu_platform() function
108 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform(); in CheckIfTPUInitialized() local
4412 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in GetDeviceTopology() local