diff options
author | 2018-06-05 14:46:22 -0700 | |
---|---|---|
committer | 2018-06-05 14:51:27 -0700 | |
commit | f0230735d1225f914d50824208cd7f84492a6dd3 (patch) | |
tree | ebbc0681bd9b720bbf06db32543c3339fe31ab76 /tensorflow/compiler/xla/tools | |
parent | d935dd9d992e9632bd2e3234fd5151a3f541f4df (diff) |
[XLA] Redesign: delete SessionModule.
PiperOrigin-RevId: 199361402
Diffstat (limited to 'tensorflow/compiler/xla/tools')
-rw-r--r-- | tensorflow/compiler/xla/tools/BUILD | 2 | ||||
-rw-r--r-- | tensorflow/compiler/xla/tools/convert_computation.cc | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/compiler/xla/tools/BUILD b/tensorflow/compiler/xla/tools/BUILD index d73bcdaf82..ff5340ee3f 100644 --- a/tensorflow/compiler/xla/tools/BUILD +++ b/tensorflow/compiler/xla/tools/BUILD @@ -135,7 +135,7 @@ tf_cc_binary( deps = [ "//tensorflow/compiler/xla:statusor", "//tensorflow/compiler/xla:types", - "//tensorflow/compiler/xla/service:session_proto", + "//tensorflow/compiler/xla/service:hlo_proto", "//tensorflow/core:lib", ], ) diff --git a/tensorflow/compiler/xla/tools/convert_computation.cc b/tensorflow/compiler/xla/tools/convert_computation.cc index fe03a6e7bd..14d01b5bfb 100644 --- a/tensorflow/compiler/xla/tools/convert_computation.cc +++ b/tensorflow/compiler/xla/tools/convert_computation.cc @@ -21,7 +21,7 @@ limitations under the License. #include <unistd.h> #include <string> -#include "tensorflow/compiler/xla/service/session.pb.h" +#include "tensorflow/compiler/xla/service/hlo.pb.h" #include "tensorflow/compiler/xla/statusor.h" #include "tensorflow/compiler/xla/types.h" #include "tensorflow/core/platform/env.h" @@ -33,7 +33,7 @@ namespace xla { namespace tools { void RealMain(const string& mode, const string& path) { - SessionModule module; + HloSnapshot module; tensorflow::Env* env = tensorflow::Env::Default(); if (mode == "txt2bin") { TF_CHECK_OK(tensorflow::ReadTextProto(env, path, &module)); |