Searched defs:session_module (Results 1 - 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | executable.cc | 144 const SessionModule& session_module) { 157 tensorflow::SerializeToStringDeterministic(session_module, &result)); 142 DumpToDirectory( const string& directory_path, string filename, const SessionModule& session_module) argument
|
H A D | 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); 49 LoadSessionModule( const SessionModule& session_module) argument
|
H A D | executable.h | 156 void set_session_module(std::unique_ptr<xla::SessionModule> session_module) { argument 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);
|
H A D | 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)); 885 executable->session_module() [all...] |
/external/tensorflow/tensorflow/compiler/xla/tools/ |
H A D | dumped_computation_to_operation_list.cc | 69 SessionModule session_module; local 71 &session_module)); 72 auto computation_status = client->LoadSnapshot(session_module);
|
H A D | 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/client/ |
H A D | local_client.cc | 183 executable_->session_module()->set_execution_platform( 185 TF_RETURN_IF_ERROR(RecordArguments(arguments, executable_->session_module())); 190 TF_RETURN_IF_ERROR(RecordResult(result.get(), executable_->session_module())); 197 SessionModule* session_module) { 198 session_module->clear_arguments(); 202 *session_module->add_arguments() = literal->ToProto(); 208 const ShapedBuffer* result, SessionModule* session_module) { 209 session_module->clear_result(); 212 *session_module->mutable_result() = literal->ToProto(); 195 RecordArguments( const tensorflow::gtl::ArraySlice<const ShapedBuffer*> arguments, SessionModule* session_module) argument 207 RecordResult( const ShapedBuffer* result, SessionModule* session_module) argument
|
/external/tensorflow/tensorflow/python/keras/_impl/keras/ |
H A D | backend.py | 31 from tensorflow.python.client import session as session_module namespace 407 _SESSION = session_module.Session(config=config) 2795 if (key not in tf_inspect.getargspec(session_module.Session.run)[0] and
|
Completed in 560 milliseconds