Adding 64 bit tensor data types (#54)

Co-authored-by: Stuart Schaefer <[email protected]>
WebAssembly · Aug 23, 2023 · 747d8df · 747d8df
1 parent c1f8b87
commit 747d8df
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/ml.md b/ml.md
@@ -26,9 +26,11 @@ Then, the user passes <em>tensor</em> inputs to the <em>graph</em>, computes the
 <ul>
 <li><a name="tensor_type.fp16"><code>FP16</code></a></li>
 <li><a name="tensor_type.fp32"><code>FP32</code></a></li>
+<li><a name="tensor_type.fp64"><code>FP64</code></a></li>
 <li><a name="tensor_type.bf16"><code>BF16</code></a></li>
 <li><a name="tensor_type.u8"><code>U8</code></a></li>
 <li><a name="tensor_type.i32"><code>I32</code></a></li>
+<li><a name="tensor_type.i64"><code>I64</code></a></li>
 </ul>
 <h4><a name="tensor_dimensions"><code>type tensor-dimensions</code></a></h4>
 <p><a href="#tensor_dimensions"><a href="#tensor_dimensions"><code>tensor-dimensions</code></a></a></p>

diff --git 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
@@ -27,9 +27,11 @@ interface tensor {
     enum tensor-type {
         FP16,
         FP32,
+        FP64,
         BF16,
         U8,
-        I32
+        I32,
+        I64
     }
 
     /// The tensor data.