diff options
Diffstat (limited to 'tests/SimpleTensor.h')
| -rw-r--r-- | tests/SimpleTensor.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/SimpleTensor.h b/tests/SimpleTensor.h index 9ea171d492..419621e808 100644 --- a/tests/SimpleTensor.h +++ b/tests/SimpleTensor.h @@ -392,6 +392,8 @@ int SimpleTensor<T>::num_channels() const case Format::S16: case Format::U32: case Format::S32: + case Format::U64: + case Format::S64: case Format::F16: case Format::F32: return 1; |