Skip to content

Commit

Permalink
Adding 64 bit tensor data types (#54)
Browse files Browse the repository at this point in the history
Co-authored-by: Stuart Schaefer <[email protected]>
  • Loading branch information
shschaefer and Stuart Schaefer committed Aug 23, 2023
1 parent c1f8b87 commit 747d8df
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 1 deletion.
2 changes: 2 additions & 0 deletions ml.md
Original file line number Diff line number Diff line change
Expand Up @@ -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>
Expand Down
2 changes: 2 additions & 0 deletions wasi-nn.witx
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,10 @@
(enum (@witx tag u8)
$f16
$f32
$f64
$u8
$i32
$i64
)
)
(typename $tensor_data (list u8))
Expand Down
4 changes: 3 additions & 1 deletion wit/wasi-nn.wit
Original file line number Diff line number Diff line change
Expand Up @@ -27,9 +27,11 @@ interface tensor {
enum tensor-type {
FP16,
FP32,
FP64,
BF16,
U8,
I32
I32,
I64
}

/// The tensor data.
Expand Down

0 comments on commit 747d8df

Please sign in to comment.