diff options
author | Fabrizio Indirli <Fabrizio.Indirli@arm.com> | 2023-12-11 11:15:32 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-11 19:09:55 +0000 |
commit | 7203835d65afb8a06a5cf98072718d93a9b71567 (patch) | |
tree | 7e625e5489eee34e5b6ec2d24e00102799e57344 /reference_model/src/model_runner_impl.h | |
parent | 48ed6cf8eb0d6beab4cb97f08fc41e037bfd182e (diff) | |
download | reference_model-7203835d65afb8a06a5cf98072718d93a9b71567.tar.gz |
Add support for precise mode in eager runner
Add support for Fp64 tensors in the eager runner's
helper functions, when precise mode is enabled.
Signed-off-by: Fabrizio Indirli <Fabrizio.Indirli@arm.com>
Change-Id: Ib737c0d18fb1c7ac40ce6ea03a4fbcefae88ba5c
Diffstat (limited to 'reference_model/src/model_runner_impl.h')
-rw-r--r-- | reference_model/src/model_runner_impl.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/reference_model/src/model_runner_impl.h b/reference_model/src/model_runner_impl.h index aed8a1e..db9755c 100644 --- a/reference_model/src/model_runner_impl.h +++ b/reference_model/src/model_runner_impl.h @@ -1,5 +1,5 @@ -// Copyright (c) 2022-2023, ARM Limited. +// Copyright (c) 2022-2024, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -64,6 +64,7 @@ private: GraphStatus initialize(TosaSerializationBasicBlock* bb, TosaSerializationHandler* serialization_handler); void validateTosaVersion(TosaSerializationHandler& serialization_handler); void checkGraphStatus(SubgraphTraverser& main_gt); + int setInputForPrecMode(Tensor* tensor, std::string input_name, uint8_t* raw_ptr, size_t size); }; }; // namespace TosaReference |