Searched defs:session_module (Results 1 - 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.cc144 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 Dcomputation_tracker.cc50 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 Dexecutable.h156 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 Dservice.cc356 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 Ddumped_computation_to_operation_list.cc69 SessionModule session_module; local
71 &session_module));
72 auto computation_status = client->LoadSnapshot(session_module);
H A Ddumped_computation_to_text.cc43 SessionModule session_module; local
45 &session_module));
46 auto computation_status = client->LoadSnapshot(session_module);
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dlocal_client.cc183 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 Dbackend.py31 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 644 milliseconds