diff options
Diffstat (limited to 'arm_compute/graph/mutators/SyntheticDataTypeMutator.h')
-rw-r--r-- | arm_compute/graph/mutators/SyntheticDataTypeMutator.h | 7 |
1 files changed, 6 insertions, 1 deletions
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 |