diff options
Diffstat (limited to 'reference_model/src/tensor.h')
| -rw-r--r-- | reference_model/src/tensor.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/tensor.h b/reference_model/src/tensor.h index 4efbf84..a3ce4bb 100644 --- a/reference_model/src/tensor.h +++ b/reference_model/src/tensor.h @@ -646,6 +646,7 @@ public: { case DType_FP32: case DType_FP16: + case DType_BF16: switch (rank) { case 0: |