Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtpu_client.h73 StatusOr<std::string> SerializeExecutable(
Dtpu_client.cc140 StatusOr<std::string> PjRtTpuClient::SerializeExecutable( in SerializeExecutable() function in xla::PjRtTpuClient
Dpjrt_client.h226 virtual StatusOr<std::string> SerializeExecutable(
Dtfrt_cpu_pjrt_client.h139 StatusOr<std::string> SerializeExecutable( in SerializeExecutable() function
Dpjrt_stream_executor_client.h180 StatusOr<std::string> SerializeExecutable( in SerializeExecutable() function
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compilation_cache_rpc_support.cc100 auto get_executable_status = tpu_program_group->SerializeExecutable( in SerializeCacheEntryToBufferSlices()
Dtpu_program_group.h151 Status SerializeExecutable(int index,
Dtpu_program_group.cc371 Status TpuProgramGroup::SerializeExecutable( in SerializeExecutable() function in tensorflow::tpu::TpuProgramGroup
/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_client.h155 StatusOr<pybind11::bytes> SerializeExecutable(
Dpy_client.cc208 StatusOr<py::bytes> PyClient::SerializeExecutable( in SerializeExecutable() function in xla::PyClient
210 return pjrt_client_->SerializeExecutable(executable.pjrt_executable()); in SerializeExecutable()
Dxla.cc240 .def("serialize_executable", &PyClient::SerializeExecutable) in PYBIND11_MODULE()