diff options
author | Kevin Cheng <kevin.cheng@arm.com> | 2021-01-22 17:21:02 -0800 |
---|---|---|
committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-02-01 14:19:58 -0800 |
commit | 3a4785772a0f076e2e9653ad70dbb2fc412ac302 (patch) | |
tree | 5260a30572f8359421a410ba4f2cc7512e1e773c /reference_model/src/ops/tensor_ops.cc | |
parent | e2e13cda4c71f9d72935107bbea9a97df779e347 (diff) | |
download | reference_model-3a4785772a0f076e2e9653ad70dbb2fc412ac302.tar.gz |
Remove AINT8
- Updated C and Py serialization libraries, updated licence files
- Removed AINT8 from TOSA reference tests
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Signed-off-by: Jared Smolens <jared.smolens@arm.com>
Change-Id: I860bfeaad5a075e50f569c8f6861927ebacf1378
Diffstat (limited to 'reference_model/src/ops/tensor_ops.cc')
-rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 30 |
1 files changed, 13 insertions, 17 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 82ce3d2..d6cd1cd 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -1,5 +1,5 @@ -// Copyright (c) 2020, ARM Limited. +// Copyright (c) 2020-2021, ARM Limited. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. @@ -1189,41 +1189,37 @@ int OpTransposeConv2d<InDtype, OutDtype>::eval() // template explicit instantiation DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, FLOAT); -DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, AINT8); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, INT8); DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpArgMax, INT16); DEF_INSTANTIATE_ONE_TYPE(OpAvgPool2d, FLOAT) -DEF_INSTANTIATE_ONE_TYPE(OpAvgPool2d, AINT8) +DEF_INSTANTIATE_ONE_TYPE(OpAvgPool2d, INT8) DEF_INSTANTIATE_ONE_TYPE(OpAvgPool2d, INT16) DEF_INSTANTIATE_TWO_TYPE(OpConv2d, FLOAT, FLOAT); -DEF_INSTANTIATE_TWO_TYPE(OpConv2d, AINT8, INT4); -DEF_INSTANTIATE_TWO_TYPE(OpConv2d, AINT8, INT8); -DEF_INSTANTIATE_TWO_TYPE(OpConv2d, AINT8, AINT8); +DEF_INSTANTIATE_TWO_TYPE(OpConv2d, INT8, INT4); +DEF_INSTANTIATE_TWO_TYPE(OpConv2d, INT8, INT8); DEF_INSTANTIATE_TWO_TYPE(OpConv2d, INT16, INT8); DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, FLOAT, FLOAT); -DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, AINT8, INT4); -DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, AINT8, INT8); -DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, AINT8, AINT8); +DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, INT8, INT4); +DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, INT8, INT8); DEF_INSTANTIATE_TWO_TYPE(OpDepthwiseConv2d, INT16, INT8); DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, FLOAT, FLOAT); -DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, AINT8, INT4); -DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, AINT8, INT8); -DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, AINT8, AINT8); +DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, INT8, INT4); +DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, INT8, INT8); DEF_INSTANTIATE_TWO_TYPE(OpFullyConnected, INT16, INT8); -DEF_INSTANTIATE_ONE_TYPE(OpMatMul, AINT8); +DEF_INSTANTIATE_ONE_TYPE(OpMatMul, INT8); DEF_INSTANTIATE_ONE_TYPE(OpMatMul, INT16); DEF_INSTANTIATE_ONE_TYPE(OpMatMul, FLOAT); DEF_INSTANTIATE_ONE_TYPE(OpMaxPool2d, FLOAT); -DEF_INSTANTIATE_ONE_TYPE(OpMaxPool2d, AINT8); +DEF_INSTANTIATE_ONE_TYPE(OpMaxPool2d, INT8); DEF_INSTANTIATE_ONE_TYPE(OpMaxPool2d, INT16); DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, FLOAT, FLOAT); -DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, AINT8, INT4); -DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, AINT8, INT8); -DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, AINT8, AINT8); +DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, INT8, INT4); +DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, INT8, INT8); DEF_INSTANTIATE_TWO_TYPE(OpTransposeConv2d, INT16, INT8); |