HomeSort by relevance Sort by last modified time
    Searched defs:session_module (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tools/
dumped_computation_to_operation_list.cc 69 SessionModule session_module; local
71 &session_module));
72 auto computation_status = client->LoadSnapshot(session_module);
dumped_computation_to_text.cc 43 SessionModule session_module; local
45 &session_module));
46 auto computation_status = client->LoadSnapshot(session_module);
  /external/tensorflow/tensorflow/compiler/xla/service/
executable.h 156 void set_session_module(std::unique_ptr<xla::SessionModule> session_module) {
157 session_module_ = std::move(session_module);
160 SessionModule* session_module() const { return session_module_.get(); } function in class:xla::Executable
163 // Dump session_module to directory_path/filename.
165 const SessionModule& session_module);
computation_tracker.cc 50 const SessionModule& session_module) {
62 session_module.embedded_computations()) {
72 session_module.embedded_computations()) {
82 const int64 old_handle = session_module.entry().computation_handle().handle();
85 LoadSessionComputation(session_module.entry(), &old_to_new));
100 auto session_module = MakeUnique<SessionModule>(); local
101 *session_module->mutable_entry() =
106 *session_module->add_embedded_computations() =
109 return std::move(session_module);
service.cc 356 std::unique_ptr<SessionModule> session_module,
361 session_module->entry().name().c_str(),
364 *session_module));
365 session_modules.push_back(std::move(session_module));
408 std::unique_ptr<SessionModule> session_module; local
415 session_module,
420 session_module->entry().name().c_str(),
423 *session_module));
444 executable->set_session_module(std::move(session_module));
    [all...]

Completed in 126 milliseconds