Home
last modified time | relevance | path

Searched refs:GetHloModules (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/
Dpy_executable.cc187 return executable_->GetHloModules(); in HloModules()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_client.h454 virtual StatusOr<std::vector<std::shared_ptr<HloModule>>> GetHloModules()
Dpjrt_stream_executor_client.h647 StatusOr<std::vector<std::shared_ptr<HloModule>>> GetHloModules()
Dpjrt_stream_executor_client.cc2103 PjRtStreamExecutorExecutable::GetHloModules() const { in GetHloModules() function in xla::PjRtStreamExecutorExecutable