blob: 49f03e94110247481062f782feed074c5a8e3ce7 [file] [log] [blame]
Eric Kunzee5e26762020-10-13 16:11:07 -07001
Eric Kunze286f8342022-06-22 11:30:23 -07002// Copyright (c) 2020-2022, ARM Limited.
Eric Kunzee5e26762020-10-13 16:11:07 -07003//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8// http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15
16#ifndef FUNC_CONFIG_H_
17#define FUNC_CONFIG_H_
18
Eric Kunzee5e26762020-10-13 16:11:07 -070019struct func_config_t
20{
Eric Kunze286f8342022-06-22 11:30:23 -070021 std::string operator_fbs = "tosa.fbs";
22 std::string test_desc = "desc.json";
23 std::string flatbuffer_dir = "";
24 std::string output_dir = "";
25 std::string tosa_file = "";
26 std::string ifm_name = "";
27 std::string ifm_file = "";
28 std::string ofm_name = "";
29 std::string ofm_file = "";
30 uint32_t eval = 1;
31 uint32_t validate_only = 0;
32 uint32_t output_tensors = 1;
33 uint32_t tosa_profile = 1;
34 uint32_t dump_intermediates = 0;
35 std::string fp_format = "0.5";
Eric Kunzee5e26762020-10-13 16:11:07 -070036};
37
38// Forward declaration
39struct func_debug_t;
40
Eric Kunze286f8342022-06-22 11:30:23 -070041int func_model_parse_cmd_line(
42 func_config_t& func_config, func_debug_t& func_debug, int argc, char** argv, const char* version);
Eric Kunzee5e26762020-10-13 16:11:07 -070043int func_model_parse_flat_config_file(func_config_t*, const char* filename);
Eric Kunze286f8342022-06-22 11:30:23 -070044void func_model_print_help();
Eric Kunzee5e26762020-10-13 16:11:07 -070045
46#endif