diff options
Diffstat (limited to 'tensorflow/compiler/aot/tfcompile_main.cc')
| -rw-r--r-- | tensorflow/compiler/aot/tfcompile_main.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/compiler/aot/tfcompile_main.cc b/tensorflow/compiler/aot/tfcompile_main.cc index cc499c3284..6ab3d47418 100644 --- a/tensorflow/compiler/aot/tfcompile_main.cc +++ b/tensorflow/compiler/aot/tfcompile_main.cc @@ -94,6 +94,8 @@ Status Main(const MainFlags& flags) { TF_RETURN_IF_ERROR(WriteStringToFile(env, flags.out_object, StringPiece(obj.data(), obj.size()))); HeaderOpts header_opts; + header_opts.gen_name_to_index = flags.gen_name_to_index; + header_opts.gen_program_shape = flags.gen_program_shape; if (flags.cpp_class.empty()) { return errors::InvalidArgument("Must specify --cpp_class"); } |