Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.h220 virtual std::unique_ptr<LoadedProgramHandle> LoadProgram(
Drecording_tpu_driver.cc361 std::unique_ptr<LoadedProgramHandle> LoadProgram( in LoadProgram() function in tpu_driver::__anone63593b20111::RecordingTpuDriver
370 std::make_unique<RecordingLoadedProgramHandle>(driver_->LoadProgram( in LoadProgram()
Dgrpc_tpu_driver.cc201 std::unique_ptr<LoadedProgramHandle> LoadProgram(
397 std::unique_ptr<LoadedProgramHandle> LoadProgram( in LoadProgram() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver
400 return streams_[core_id]->LoadProgram(core_id, handle, wait_for); in LoadProgram()
861 std::unique_ptr<LoadedProgramHandle> GrpcTpuStream::LoadProgram( in LoadProgram() function in tpu_driver::__anon4073f9c10111::GrpcTpuStream
Dpod_tpu_driver.cc592 std::unique_ptr<LoadedProgramHandle> LoadProgram( in LoadProgram() function in tpu_driver::__anon6b014a980111::PodTpuDriver
611 core_to_driver_[core_id]->LoadProgram( in LoadProgram()
Ddirect_tpu_driver.cc400 std::unique_ptr<LoadedProgramHandle> LoadProgram( in LoadProgram() function in tpu_driver::__anon9d833a900111::DirectTpuDriver
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc533 client_->driver()->LoadProgram(device_id, compiled_program.get(), {}); in PyTpuExecutable()