aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/model_runner_impl.h
diff options
context:
space:
mode:
authorFabrizio Indirli <Fabrizio.Indirli@arm.com>2023-12-11 11:15:32 +0000
committerEric Kunze <eric.kunze@arm.com>2024-01-11 19:09:55 +0000
commit7203835d65afb8a06a5cf98072718d93a9b71567 (patch)
tree7e625e5489eee34e5b6ec2d24e00102799e57344 /reference_model/src/model_runner_impl.h
parent48ed6cf8eb0d6beab4cb97f08fc41e037bfd182e (diff)
downloadreference_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.h3
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