Searched refs:TpuExecutableInterface (Results 1 – 6 of 6) sorted by relevance
41 class TpuExecutableInterface : public Executable {43 explicit TpuExecutableInterface(std::shared_ptr<HloModule> hlo_module) in TpuExecutableInterface() function45 ~TpuExecutableInterface() override = default;
36 class TpuExecutable : public TpuExecutableInterface {
54 TpuExecutableInterface::AllocateOutputMemoryWithInputReuse( in AllocateOutputMemoryWithInputReuse()168 StatusOr<ExecutionOutput> TpuExecutableInterface::ExecuteAsyncOnStream( in ExecuteAsyncOnStream()
33 : TpuExecutableInterface(std::move(hlo_module)), in TpuExecutable()
127 xla::TpuExecutableInterface* tpu_executable = in ExecutableFingerprint()128 tensorflow::down_cast<xla::TpuExecutableInterface*>( in ExecutableFingerprint()
88 class TpuExecutable : public TpuExecutableInterface {92 : TpuExecutableInterface(std::move(hlo_module)), in TpuExecutable()