diff options
| author | James Ward <james.ward@arm.com> | 2022-11-15 11:36:47 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2022-11-29 15:55:26 +0000 |
| commit | ee2566914d3476b8103b88915f3b81bda8490b44 (patch) | |
| tree | 0d0dd56adafb3a65d896a192eeb736200eec8f06 /reference_model/src/ops/op_factory.h | |
| parent | 542dd3b8da39440026fa9e809eebd0a3b79cf95d (diff) | |
| download | reference_model-ee2566914d3476b8103b88915f3b81bda8490b44.tar.gz | |
FP16 improvements
* Update FP16 resize to newest spec version
* Correct casting to fp16 for graphs of >1 ops
Change-Id: Iedff9a71eb7f72948b3c00a635bb0fd07d414bcd
Signed-off-by: James Ward <james.ward@arm.com>
Diffstat (limited to 'reference_model/src/ops/op_factory.h')
| -rw-r--r-- | reference_model/src/ops/op_factory.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/reference_model/src/ops/op_factory.h b/reference_model/src/ops/op_factory.h index f399bd1..f4177db 100644 --- a/reference_model/src/ops/op_factory.h +++ b/reference_model/src/ops/op_factory.h @@ -108,7 +108,7 @@ #define DEF_FACTORY_TWO_TYPE_RESIZE_FP16(OP, DTYPE1, DTYPE2) \ if (inputDType == DType_##DTYPE1 && outputDType == DType_##DTYPE2) \ { \ - return new OP<DType_##DTYPE1, DType_##DTYPE2, float>(sgt, attribute, id); \ + return new OP<DType_##DTYPE1, DType_##DTYPE2, half_float::half>(sgt, attribute, id); \ } #define DEF_FACTORY_TWO_TYPE_RESIZE_BF16(OP, DTYPE1, DTYPE2) \ |