diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2019-12-02 11:58:19 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2019-12-06 14:54:19 +0000 |
commit | f4261adf78bdb9f8b2d6f2970636125096c173cb (patch) | |
tree | a2cf7abd84787720bdc286f09422336691d6fa95 /examples/graph_srcnn955.cpp | |
parent | 6f58b37a18cfade5dbec38638926f7bd368756d9 (diff) | |
download | ComputeLibrary-f4261adf78bdb9f8b2d6f2970636125096c173cb.tar.gz |
COMPMID-2779: Add support for generating synthetic int8 graphs.
Adds SyntheticDataTypeMutator, which is responsible for mutating graphs
to int8 and thus enable performance analysis on a wider range of models.
Change-Id: I9a00f0ae59421ab11952660f5115b5dcd9314aaf
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2418
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'examples/graph_srcnn955.cpp')
-rw-r--r-- | examples/graph_srcnn955.cpp | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/examples/graph_srcnn955.cpp b/examples/graph_srcnn955.cpp index b69305889b..29faf72ac2 100644 --- a/examples/graph_srcnn955.cpp +++ b/examples/graph_srcnn955.cpp @@ -75,9 +75,6 @@ public: std::cout << "Image width: " << image_width << std::endl; std::cout << "Image height: " << image_height << std::endl; - // Checks - ARM_COMPUTE_EXIT_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "QASYMM8 not supported for this graph"); - // Get trainable parameters data path const std::string data_path = common_params.data_path; const std::string model_path = "/cnn_data/srcnn955_model/"; @@ -120,10 +117,11 @@ public: // Finalize graph GraphConfig config; - config.num_threads = common_params.threads; - config.use_tuner = common_params.enable_tuner; - config.tuner_mode = common_params.tuner_mode; - config.tuner_file = common_params.tuner_file; + config.num_threads = common_params.threads; + config.use_tuner = common_params.enable_tuner; + config.tuner_mode = common_params.tuner_mode; + config.tuner_file = common_params.tuner_file; + config.convert_to_uint8 = (common_params.data_type == DataType::QASYMM8); graph.finalize(common_params.target, config); |