Searched refs:tpu_platform (Results 1 – 7 of 7) sorted by relevance
20 REGISTER_MODULE_INITIALIZER(tpu_platform,28 REGISTER_MODULE_INITIALIZER_SEQUENCE(tpu_platform, multi_platform_manager);30 tpu_platform);
97 tpu_platform().mutex().lock(); in DeallocateStream()99 tpu_platform().mutex().unlock(); in DeallocateStream()111 tpu_platform().EraseEvent(event->implementation()); in DeallocateEvent()134 auto se_event = tpu_platform().LookupEvent(event->implementation()); in PollForEventStatus()143 auto se_event = tpu_platform().LookupEvent(event->implementation()); in RecordEvent()153 auto se_event = tpu_platform().LookupEvent(event->implementation()); in WaitForEvent()180 tpu_platform().mutex().lock(); in GetStreamImplementation()182 tpu_platform().mutex().unlock(); in GetStreamImplementation()191 tpu_platform().InsertEvent(ptr.get(), tpu_event); in CreateEventImplementation()
106 "tpu_platform.h",131 hdrs = ["tpu_platform.h"],147 "tpu_platform.h",178 "tpu_platform.cc",183 "tpu_platform.h",
223 TpuPlatform& tpu_platform() { in tpu_platform() function228 return *(tpu_platform().stream_map()); in stream_map()232 tensorflow::mutex_lock m(tpu_platform().mutex()); in get_stream()
74 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in RunTransferWithOrdinal() local84 tpu_platform->ExecutorForDevice(real_device_ordinal)); in RunTransferWithOrdinal()
105 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform(); in CheckIfTPUInitialized() local106 if (!tpu_platform->Initialized()) { in CheckIfTPUInitialized()
3957 auto* tpu_platform = tpu::TpuPlatformInterface::GetRegisteredPlatform( in GetDeviceTopology() local3959 TF_RET_CHECK(tpu_platform); in GetDeviceTopology()3960 tpu::TpuTopologyExternal tpu_topology(tpu_platform->GetTopologyPtr()); in GetDeviceTopology()