Searched refs:TpuExecutableInterface (Results 1 – 7 of 7) sorted by relevance
41 class TpuExecutableInterface : public Executable {43 explicit TpuExecutableInterface(std::shared_ptr<HloModule> hlo_module) in TpuExecutableInterface() function45 ~TpuExecutableInterface() override = default;
24 class TpuExecutable : public xla::TpuExecutableInterface {28 : TpuExecutableInterface(std::move(hlo_module)), in TpuExecutable()
54 TpuExecutableInterface::AllocateOutputMemoryWithInputReuse( in AllocateOutputMemoryWithInputReuse()183 StatusOr<ExecutionOutput> TpuExecutableInterface::ExecuteAsyncOnStream( in ExecuteAsyncOnStream()
36 class TpuOpExecutable : public xla::TpuExecutableInterface {
33 : TpuExecutableInterface(std::move(hlo_module)), in TpuOpExecutable()
147 xla::TpuExecutableInterface* tpu_executable = in ExecutableFingerprint()148 tensorflow::down_cast<xla::TpuExecutableInterface*>( in ExecutableFingerprint()