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

type tensor-dimensions

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.