Home
last modified time | relevance | path

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

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