| /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);
|
| compile_only_service.cc | 82 std::unique_ptr<SessionModule> session_module,
86 session_module->entry().name(), "__version_",
92 *session_module));
|
| computation_tracker.h | 60 const SessionModule& session_module);
|
| executable.cc | 144 const SessionModule& session_module) {
157 tensorflow::SerializeToStringDeterministic(session_module, &result));
|
| 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...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| 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();
|
| local_client.h | 84 SessionModule* session_module);
88 SessionModule* session_module);
|
| /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/python/keras/_impl/keras/ |
| backend.py | 31 from tensorflow.python.client import session as session_module
407 _SESSION = session_module.Session(config=config)
[all...] |
