diff options
Diffstat (limited to 'reference_model/src/ops')
| -rw-r--r-- | reference_model/src/ops/ewise_binary.cc | 1 | ||||
| -rw-r--r-- | reference_model/src/ops/reduction.cc | 9 |
2 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/ops/ewise_binary.cc b/reference_model/src/ops/ewise_binary.cc index 2bc894d..8578527 100644 --- a/reference_model/src/ops/ewise_binary.cc +++ b/reference_model/src/ops/ewise_binary.cc @@ -654,6 +654,7 @@ DEF_INSTANTIATE_RANK0_6_ONE_RANK_TWO_TYPE(BinaryNodeBase, BF16, BOOL); DEF_INSTANTIATE_RANK0_6_ONE_RANK_TWO_TYPE(BinaryNodeBase, FP32, BOOL); DEF_INSTANTIATE_RANK0_6_ONE_RANK_TWO_TYPE(BinaryNodeBase, INT32, BOOL); DEF_INSTANTIATE_RANK0_6_ONE_RANK_TWO_TYPE(BinaryNodeBase, FP64, FP64); +DEF_INSTANTIATE_RANK0_6_ONE_RANK_TWO_TYPE(BinaryNodeBase, FP64, BOOL); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpAdd, FP16); DEF_INSTANTIATE_RANK0_6_ONE_RANK_ONE_TYPE(OpAdd, BF16); diff --git a/reference_model/src/ops/reduction.cc b/reference_model/src/ops/reduction.cc index f07ffd7..639da8d 100644 --- a/reference_model/src/ops/reduction.cc +++ b/reference_model/src/ops/reduction.cc @@ -305,3 +305,12 @@ DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceSum, BF16); DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceSum, FP32); DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceSumDouble, FP64); DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(OpReduceSumInt, INT32); + +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, BOOL); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, INT8); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, INT16); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, INT32); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, FP16); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, BF16); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, FP32); +DEF_INSTANTIATE_RANK1_6_ONE_RANK_ONE_TYPE(ReduceNode, FP64);
\ No newline at end of file |