diff options
Diffstat (limited to 'arm_compute/graph')
-rw-r--r-- | arm_compute/graph/Types.h | 4 | ||||
-rw-r--r-- | arm_compute/graph/mutators/SyntheticDataTypeMutator.h | 7 |
2 files changed, 9 insertions, 2 deletions
diff --git a/arm_compute/graph/Types.h b/arm_compute/graph/Types.h index 77e91b205a..c4afba2dee 100644 --- a/arm_compute/graph/Types.h +++ b/arm_compute/graph/Types.h @@ -76,6 +76,7 @@ constexpr EdgeID EmptyEdgeID = std::numeric_limits<EdgeID>::max(); // Forward declarations struct TensorDescriptor; + /** Graph configuration structure */ struct GraphConfig { @@ -83,7 +84,8 @@ struct GraphConfig bool use_function_weights_manager{ true }; /**< Use a weights manager to manage transformed weights */ bool use_transition_memory_manager{ true }; /**< Use a memory manager to manager transition buffer memory */ bool use_tuner{ false }; /**< Use a tuner in tunable backends */ - bool convert_to_uint8{ false }; /**< Convert graph to a synthetic uint8 graph */ + bool use_synthetic_type{ false }; /**< Convert graph to a synthetic graph for a data type */ + DataType synthetic_type{ DataType::QASYMM8 }; /**< The data type of the synthetic graph */ CLTunerMode tuner_mode{ CLTunerMode::EXHAUSTIVE }; /**< Tuner mode to be used by the CL tuner */ int num_threads{ -1 }; /**< Number of threads to use (thread capable backends), if 0 the backend will auto-initialize, if -1 the backend will stay as it is. */ std::string tuner_file{ "acl_tuner.csv" }; /**< File to load/store tuning values from */ diff --git a/arm_compute/graph/mutators/SyntheticDataTypeMutator.h b/arm_compute/graph/mutators/SyntheticDataTypeMutator.h index ed270f894f..2292e52086 100644 --- a/arm_compute/graph/mutators/SyntheticDataTypeMutator.h +++ b/arm_compute/graph/mutators/SyntheticDataTypeMutator.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019 Arm Limited. + * Copyright (c) 2019-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -35,10 +35,15 @@ namespace graph class SyntheticDataTypeMutator final : public IGraphMutator { public: + // Constructor + SyntheticDataTypeMutator(DataType mutate_type = DataType::QASYMM8); // Inherited methods overridden virtual void mutate(Graph &g) override; MutationType type() const override; const char *name() override; + +private: + DataType _mutate_type; }; } // namespace graph } // namespace arm_compute |