Skip to content

Commit ae2ec68

Browse files
authored
add pytorch flag (#25)
* add pytorch flag Signed-off-by: Jianbai Ye <[email protected]> * update abi.md for pytorch flag Signed-off-by: Jianbai Ye <[email protected]>
1 parent 0c66d0d commit ae2ec68

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

wasi-nn.abi.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,9 @@ Size: 1, Alignment: 1
7171
- <a href="graph_encoding.tensorflow" name="graph_encoding.tensorflow"></a> [`tensorflow`](#graph_encoding.tensorflow)
7272

7373

74+
- <a href="graph_encoding.pytorch" name="graph_encoding.pytorch"></a> [`pytorch`](#graph_encoding.pytorch)
75+
76+
7477
## <a href="#execution_target" name="execution_target"></a> `execution-target`: enum
7578

7679

wasi-nn.wit.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,8 @@ resource graph
7575
enum graph-encoding {
7676
openvino,
7777
onnx,
78-
tensorflow
78+
tensorflow,
79+
pytorch
7980
}
8081
8182
// Define where the graph should be executed.

0 commit comments

Comments
 (0)