diff --git a/ml.md b/ml.md
index 3a8dcc3..fad73be 100644
--- a/ml.md
+++ b/ml.md
@@ -26,9 +26,11 @@ Then, the user passes tensor inputs to the graph, computes the
tensor-dimensions
diff --git a/wasi-nn.witx b/wasi-nn.witx
index 30f921b..e413c2e 100644
--- a/wasi-nn.witx
+++ b/wasi-nn.witx
@@ -20,8 +20,10 @@
(enum (@witx tag u8)
$f16
$f32
+ $f64
$u8
$i32
+ $i64
)
)
(typename $tensor_data (list u8))
diff --git a/wit/wasi-nn.wit b/wit/wasi-nn.wit
index 2bcf909..6b412ac 100644
--- a/wit/wasi-nn.wit
+++ b/wit/wasi-nn.wit
@@ -27,9 +27,11 @@ interface tensor {
enum tensor-type {
FP16,
FP32,
+ FP64,
BF16,
U8,
- I32
+ I32,
+ I64
}
/// The tensor data.