Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable_interface.h41 class TpuExecutableInterface : public Executable {
43 explicit TpuExecutableInterface(std::shared_ptr<HloModule> hlo_module) in TpuExecutableInterface() function
45 ~TpuExecutableInterface() override = default;
H A Dtpu_executable.h24 class TpuExecutable : public xla::TpuExecutableInterface {
28 : TpuExecutableInterface(std::move(hlo_module)), in TpuExecutable()
H A Dtpu_executable_interface.cc54 TpuExecutableInterface::AllocateOutputMemoryWithInputReuse( in AllocateOutputMemoryWithInputReuse()
183 StatusOr<ExecutionOutput> TpuExecutableInterface::ExecuteAsyncOnStream( in ExecuteAsyncOnStream()
H A Dtpu_op_executable.h36 class TpuOpExecutable : public xla::TpuExecutableInterface {
H A Dtpu_op_executable.cc33 : TpuExecutableInterface(std::move(hlo_module)), in TpuOpExecutable()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.cc147 xla::TpuExecutableInterface* tpu_executable = in ExecutableFingerprint()
148 tensorflow::down_cast<xla::TpuExecutableInterface*>( in ExecutableFingerprint()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/tpu/
H A Dtpu_op_executable.cc33 : TpuExecutableInterface(std::move(hlo_module)), in TpuOpExecutable()