Searched refs:TpuExecutor (Results 1 – 5 of 5) sorted by relevance
35 TpuExecutor::~TpuExecutor() { in ~TpuExecutor()39 Status TpuExecutor::Init(int device_ordinal, in Init()51 int TpuExecutor::PlatformDeviceCount() { in PlatformDeviceCount()55 void TpuExecutor::SyncAndForgetFailedStreams() { in SyncAndForgetFailedStreams()59 bool TpuExecutor::SynchronizeAllActivity() { in SynchronizeAllActivity()63 Status TpuExecutor::BlockHostUntilDone(Stream* stream) { in BlockHostUntilDone()70 Status TpuExecutor::BlockUntilDoneOrFailed() { in BlockUntilDoneOrFailed()77 Status TpuExecutor::GetStatus(Stream* stream) { in GetStatus()84 tpu::TpuCoreLocationExternal TpuExecutor::GetCoreLocationExternal() const { in GetCoreLocationExternal()89 bool TpuExecutor::AllocateStream(Stream* stream) { in AllocateStream()[all …]
95 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralToInfeed()109 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferBuffersToInfeed()135 auto* tpu_executor = static_cast<TpuExecutor*>(executor->implementation()); in TransferLiteralFromOutfeed()156 se.push_back(static_cast<TpuExecutor*>(executor[i]->implementation()) in ResetDevices()246 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanShapedBufferBeAccessedNow()259 auto* tpu_executor = down_cast<TpuExecutor*>(executor->implementation()); in CanBufferBeAccessedNow()
42 class TpuExecutor : public tensorflow::tpu::TpuExecutorInterface {60 explicit TpuExecutor(::tensorflow::tpu::TpuPlatformInterface* platform, in TpuExecutor() function64 ~TpuExecutor() override;
104 this, std::make_unique<TpuExecutor>(this, executor), config.ordinal); in GetUncachedExecutor()
234 static_cast<tensorflow::tpu::TpuExecutor*>(executor->implementation()) in RunHloPasses()263 static_cast<tensorflow::tpu::TpuExecutor*>(executor->implementation()) in RunBackend()303 se_lists[i].exec[j] = static_cast<tensorflow::tpu::TpuExecutor*>( in Compile()