Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc154 class GrpcLoadedProgramHandle : public LoadedProgramHandle { class
156 explicit GrpcLoadedProgramHandle(EventId id, std::shared_ptr<GrpcEvent> event) in GrpcLoadedProgramHandle() function in tpu_driver::__anon4073f9c10111::GrpcLoadedProgramHandle
406 static_cast<const GrpcLoadedProgramHandle*>(handle.get())->stream(); in UnloadProgram()
415 static_cast<const GrpcLoadedProgramHandle*>(program)->stream(); in ExecuteProgram()
873 return absl::make_unique<GrpcLoadedProgramHandle>(event->id(), in LoadProgram()
880 return absl::make_unique<GrpcLoadedProgramHandle>(event->id(), in LoadProgram()
891 static_cast<GrpcLoadedProgramHandle*>(handle.get())->id().AsInt()); in UnloadProgram()
905 auto program_handle = static_cast<GrpcLoadedProgramHandle*>(program); in ExecuteProgram()