Searched refs:TpuExecutable (Results 1 – 4 of 4) sorted by relevance
36 class TpuExecutable : public TpuExecutableInterface {42 explicit TpuExecutable(const XLA_TpuProgram* core_program,45 ~TpuExecutable() override = default;67 TF_DISALLOW_COPY_AND_ASSIGN(TpuExecutable);
30 TpuExecutable::TpuExecutable(const XLA_TpuProgram* core_program, in TpuExecutable() function in xla::TpuExecutable37 Status TpuExecutable::LoadProgramAndEnqueueToStream( in LoadProgramAndEnqueueToStream()106 Shape TpuExecutable::HostShapeToDeviceShape(const Shape& host_shape) { in HostShapeToDeviceShape()118 int64 TpuExecutable::ShapeSize(const Shape& shape) { in ShapeSize()127 absl::string_view TpuExecutable::fingerprint() const { in fingerprint()
88 class TpuExecutable : public TpuExecutableInterface { class90 TpuExecutable(SE_Executable* se_executable, in TpuExecutable() function in xla::__anoncef202870111::TpuExecutable95 ~TpuExecutable() override { in ~TpuExecutable()271 absl::make_unique<TpuExecutable>(result, std::move(module)); in RunBackend()338 executables.emplace_back(absl::make_unique<TpuExecutable>( in Compile()
69 using HostCommmandHandler = xla::TpuExecutable::HostCommandHandler;481 auto tpu_executable = absl::make_unique<xla::TpuExecutable>( in TPUExecute()