diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-06 17:11:37 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-06 17:14:48 -0700 |
| commit | d6f107761459dfdf8773a148e11193a3512a51a6 (patch) | |
| tree | 69480ed7e2a1aab35c0ada11e098e72f4963fb11 /tensorflow/compiler/aot | |
| parent | e001f3ad84f58ace65df4e78941bc49e2ae61967 (diff) | |
Automated rollback of commit 24787842adfefe35f5a520313d775b14c29f143a
PiperOrigin-RevId: 211895566
Diffstat (limited to 'tensorflow/compiler/aot')
| -rw-r--r-- | tensorflow/compiler/aot/embedded_protocol_buffers.h | 1 | ||||
| -rw-r--r-- | tensorflow/compiler/aot/tfcompile_main.cc | 6 |
2 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/compiler/aot/embedded_protocol_buffers.h b/tensorflow/compiler/aot/embedded_protocol_buffers.h index cf5c04ac4b..bd270045e3 100644 --- a/tensorflow/compiler/aot/embedded_protocol_buffers.h +++ b/tensorflow/compiler/aot/embedded_protocol_buffers.h @@ -20,6 +20,7 @@ limitations under the License. #ifndef TENSORFLOW_COMPILER_AOT_EMBEDDED_PROTOCOL_BUFFERS_H_ #define TENSORFLOW_COMPILER_AOT_EMBEDDED_PROTOCOL_BUFFERS_H_ +#include "absl/strings/string_view.h" #include "absl/types/span.h" #include "tensorflow/compiler/xla/statusor.h" #include "tensorflow/core/platform/protobuf.h" diff --git a/tensorflow/compiler/aot/tfcompile_main.cc b/tensorflow/compiler/aot/tfcompile_main.cc index b95b063348..1c9d30d7b0 100644 --- a/tensorflow/compiler/aot/tfcompile_main.cc +++ b/tensorflow/compiler/aot/tfcompile_main.cc @@ -35,6 +35,7 @@ limitations under the License. #include "tensorflow/core/graph/graph.h" #include "tensorflow/core/graph/tensor_id.h" #include "tensorflow/core/lib/core/errors.h" +#include "tensorflow/core/lib/core/stringpiece.h" #include "tensorflow/core/lib/strings/numbers.h" #include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/init_main.h" @@ -92,9 +93,8 @@ Status Main(const MainFlags& flags) { // Write output files. Env* env = Env::Default(); const std::vector<char>& obj = compile_result.aot->object_file_data(); - TF_RETURN_IF_ERROR( - WriteStringToFile(env, flags.out_function_object, - absl::string_view(obj.data(), obj.size()))); + TF_RETURN_IF_ERROR(WriteStringToFile(env, flags.out_function_object, + StringPiece(obj.data(), obj.size()))); CodegenOpts codegen_opts; codegen_opts.gen_name_to_index = flags.gen_name_to_index; codegen_opts.gen_program_shape = flags.gen_program_shape; |