Home
last modified time | relevance | path

Searched refs:UnloadProgram (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.h223 virtual std::shared_ptr<Event> UnloadProgram(
Drecording_tpu_driver.cc386 std::shared_ptr<Event> UnloadProgram( in UnloadProgram() function in tpu_driver::__anone63593b20111::RecordingTpuDriver
395 std::make_shared<RecordingEvent>(driver_->UnloadProgram( in UnloadProgram()
Dgrpc_tpu_driver.cc204 std::shared_ptr<Event> UnloadProgram(
402 std::shared_ptr<Event> UnloadProgram( in UnloadProgram() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver
407 return stream->UnloadProgram(std::move(handle), wait_for); in UnloadProgram()
884 std::shared_ptr<Event> GrpcTpuStream::UnloadProgram( in UnloadProgram() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
Dpod_tpu_driver.cc623 std::shared_ptr<Event> UnloadProgram( in UnloadProgram() function in tpu_driver::__anon6b014a980111::PodTpuDriver
641 auto event = core_to_driver_[core_id]->UnloadProgram( in UnloadProgram()
Ddirect_tpu_driver.cc416 std::shared_ptr<Event> UnloadProgram( in UnloadProgram() function in tpu_driver::__anon9d833a900111::DirectTpuDriver
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.h303 client_->driver()->UnloadProgram(std::move(it->second), {}); in ~PyTpuExecutable()