Home
last modified time | relevance | path

Searched refs:SerializeExecutable (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_executable.h71 virtual StatusOr<std::string> SerializeExecutable() const { in SerializeExecutable() function
H A Dtpu_client.h89 StatusOr<std::string> SerializeExecutable(
H A Dpjrt_c_api_client.cc245 StatusOr<std::string> PjRtCApiClient::SerializeExecutable( in SerializeExecutable() function in xla::PjRtCApiClient
248 return wrapped_->SerializeExecutable( in SerializeExecutable()
H A Dtpu_client.cc156 StatusOr<std::string> PjRtTpuClient::SerializeExecutable( in SerializeExecutable() function in xla::PjRtTpuClient
H A Dpjrt_c_api_client.h176 StatusOr<std::string> SerializeExecutable(
H A Dtfrt_cpu_pjrt_client.h162 StatusOr<std::string> SerializeExecutable( in SerializeExecutable() function
H A Dpjrt_client.h513 virtual StatusOr<std::string> SerializeExecutable(
H A Dpjrt_stream_executor_client.h203 StatusOr<std::string> SerializeExecutable( in SerializeExecutable() function
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_compilation_cache_rpc_support.cc101 auto get_executable_status = tpu_program_group->SerializeExecutable( in SerializeCacheEntryToBufferSlices()
H A Dtpu_program_group.h148 Status SerializeExecutable(int index,
H A Dtpu_program_group.cc364 Status TpuProgramGroup::SerializeExecutable( in SerializeExecutable() function in tensorflow::tpu::TpuProgramGroup
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client.h162 StatusOr<pybind11::bytes> SerializeExecutable(
H A Dpy_client.cc357 StatusOr<py::bytes> PyClient::SerializeExecutable( in SerializeExecutable() function in xla::PyClient
359 return pjrt_client_->SerializeExecutable(executable.pjrt_executable()); in SerializeExecutable()
H A Dxla.cc236 .def("serialize_executable", &PyClient::SerializeExecutable) in PYBIND11_MODULE()