Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtpu_client.cc147 xla::TpuExecutableInterface* tpu_executable = in ExecutableFingerprint() local
153 return std::optional<std::string>(tpu_executable->fingerprint()); in ExecutableFingerprint()
165 const TpuExecutable* tpu_executable = in SerializeExecutable() local
168 return tpu_executable->Serialize(); in SerializeExecutable()
174 TF_ASSIGN_OR_RETURN(std::unique_ptr<TpuExecutable> tpu_executable, in DeserializeExecutable()
180 XlaComputation computation(tpu_executable->module().ToProto()); in DeserializeExecutable()
195 std::move(tpu_executable), client_->mutable_backend(), in DeserializeExecutable()
H A DBUILD374 "//tensorflow/stream_executor/tpu:tpu_executable",
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/tpu/
H A DBUILD215 name = "tpu_executable",
216 hdrs = ["tpu_executable.h"],
219 "//tensorflow/compiler/xla/stream_executor/tpu:tpu_executable",
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A DBUILD372 name = "tpu_executable",
373 srcs = ["tpu_executable.cc"],
374 hdrs = ["tpu_executable.h"],
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dtpu_execute.cc485 auto tpu_executable = std::make_unique<TpuOpExecutable>( in TPUExecute() local
503 tpu_executable->ExecuteAsyncOnStream(&service_run_options, in TPUExecute()
H A DBUILD365 "//tensorflow/stream_executor/tpu:tpu_executable",