Lines Matching refs:flags
23 #include "tensorflow/compiler/aot/flags.h"
65 Status Main(const MainFlags& flags) {
68 if (flags.config.empty()) {
71 TF_RETURN_IF_ERROR(ReadProtoFile(flags.config, &config));
73 if (flags.dump_fetch_nodes) {
83 if (flags.graph.empty()) {
87 TF_RETURN_IF_ERROR(ReadProtoFile(flags.graph, &graph_def));
89 TF_RETURN_IF_ERROR(CompileGraph(graph_def, config, flags, &compile_result));
94 TF_RETURN_IF_ERROR(WriteStringToFile(env, flags.out_function_object,
97 codegen_opts.gen_name_to_index = flags.gen_name_to_index;
98 codegen_opts.gen_program_shape = flags.gen_program_shape;
99 codegen_opts.target_triple = flags.target_triple;
100 if (flags.cpp_class.empty()) {
103 TF_RETURN_IF_ERROR(ParseCppClass(flags.cpp_class, &codegen_opts.class_name,
109 TF_RETURN_IF_ERROR(WriteStringToFile(env, flags.out_metadata_object,
114 TF_RETURN_IF_ERROR(WriteStringToFile(env, flags.out_header, header));
122 tensorflow::tfcompile::MainFlags flags;
123 flags.target_triple = "x86_64-pc-linux";
124 flags.out_function_object = "out_model.o";
125 flags.out_metadata_object = "out_helper.o";
126 flags.out_header = "out.h";
127 flags.entry_point = "entry";
130 AppendMainFlags(&flag_list, &flags);
134 usage += tensorflow::Flags::Usage(argv[0], flag_list);
135 bool parsed_flags_ok = tensorflow::Flags::Parse(&argc, argv, flag_list);
140 "other than flags\n\n"
142 tensorflow::Status status = tensorflow::tfcompile::Main(flags);
