diff options
author | Tai Ly <tai.ly@arm.com> | 2023-09-07 20:49:09 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-09-15 18:10:01 +0000 |
commit | cf84bc9cccbd5dc2fceae1a81c579e41be3c9a06 (patch) | |
tree | aff6bab02c36c095a62381ac8f68d185bdccbe73 /reference_model/src/command_line_utils.h | |
parent | 00f55bf46fe36bebe44e1365becbeb1e0d9e90c9 (diff) | |
download | reference_model-cf84bc9cccbd5dc2fceae1a81c579e41be3c9a06.tar.gz |
[reference_model] Support StatefulOps and the tests for CallOnceOp
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I03cb878736ccd7e1f5e1f780d7171949a19a9de2
Diffstat (limited to 'reference_model/src/command_line_utils.h')
-rw-r--r-- | reference_model/src/command_line_utils.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/reference_model/src/command_line_utils.h b/reference_model/src/command_line_utils.h index dcb0564..f8031d9 100644 --- a/reference_model/src/command_line_utils.h +++ b/reference_model/src/command_line_utils.h @@ -48,6 +48,10 @@ int func_model_parse_cmd_line( cxxopts::value<std::string>(func_config.ofm_name)) ("ofm_file", "Output tensor numpy file to be generated. Comma(,) seperated. Specifying this will overwrite the one initialized by --test_desc.", cxxopts::value<std::string>(func_config.ofm_file)) + ("variable_name", "Region tensor name. Comma(,) seperated. Specifying this will overwrite the one initialized by --test_desc.", + cxxopts::value<std::string>(func_config.variable_name)) + ("variable_file", "Region tensor numpy file to be generated. Comma(,) seperated. Specifying this will overwrite the one initialized by --test_desc.", + cxxopts::value<std::string>(func_config.variable_file)) ("eval", "Evaluate the network (0/1)", cxxopts::value<uint32_t>(func_config.eval)) ("fp_format", "Floating-point number dump format string (printf-style format, e.g. 0.5)", cxxopts::value<std::string>(func_config.fp_format)) @@ -60,6 +64,7 @@ int func_model_parse_cmd_line( cxxopts::value<tosa_level_t>(func_config.tosa_level)) ("dump_intermediates", "Dump intermediate tensors (0/1)", cxxopts::value<uint32_t>(func_config.dump_intermediates)) ("p,precise_mode", "Calculate floating point operations in FP64 (0/1)", cxxopts::value<uint32_t>(func_config.precise_mode)) + ("initialize_variable_tensor_from_numpy", "Initialize variable tensors from flatbuffer (0, default) or numpy (1)", cxxopts::value<uint32_t>(func_config.initialize_variable_tensor_from_numpy)) ("v,version", "print model version") ("i,input_tensor_file", "specify input tensor files", cxxopts::value<std::vector<std::string>>()) ("l,loglevel", func_debug.get_debug_verbosity_help_string(), cxxopts::value<std::string>()) |