diff options
Diffstat (limited to 'reference_model/src/ops/scatter_gather.h')
-rw-r--r-- | reference_model/src/ops/scatter_gather.h | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/reference_model/src/ops/scatter_gather.h b/reference_model/src/ops/scatter_gather.h index af09153..fb675a9 100644 --- a/reference_model/src/ops/scatter_gather.h +++ b/reference_model/src/ops/scatter_gather.h @@ -1,5 +1,5 @@ -// Copyright (c) 2020, ARM Limited. +// Copyright (c) 2020-2023, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -23,7 +23,7 @@ using namespace tosa; namespace TosaReference { -template <DType Dtype> +template <TOSA_REF_TYPE Dtype> class OpGather : public GraphNode { public: @@ -45,7 +45,7 @@ protected: TosaReference::TensorTemplate<TOutput>* output; }; -template <DType Dtype> +template <TOSA_REF_TYPE Dtype> class OpScatter : public GraphNode { public: |