Skip to content

Commit ebdabe9

Browse files
authored
Update README.md (#2)
Signed-off-by: Justin Chu <[email protected]>
1 parent b4789a9 commit ebdabe9

File tree

1 file changed

+22
-1
lines changed

1 file changed

+22
-1
lines changed

README.md

Lines changed: 22 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1,22 @@
1-
# onnx-ir
1+
# ONNX IR
2+
3+
An in-memory IR that supports the full ONNX spec, designed for graph construction, analysis and transformation.
4+
5+
## Features ✨
6+
7+
- Full ONNX spec support: all valid models representable by ONNX protobuf, and a subset of invalid models (so you can load and fix them).
8+
- Low memory footprint: mmap'ed external tensors; unified interface for ONNX TensorProto, Numpy arrays and PyTorch Tensors etc. No tensor size limitation. Zero copies.
9+
- Straightforward access patterns: Access value information and traverse the graph topology at ease.
10+
- Robust mutation: Create as many iterators as you like on the graph while mutating it.
11+
- Speed: Performant graph manipulation, serialization/deserialization to Protobuf.
12+
- Pythonic and familiar APIs: Classes define Pythonic apis and still map to ONNX protobuf concepts in an intuitive way.
13+
- No protobuf dependency: The IR does not require protobuf once the model is converted to the IR representation, decoupling from the serialization format.
14+
15+
## Code Organization 🗺️
16+
17+
- [`_protocols.py`](_protocols.py): Interfaces defined for all entities in the IR.
18+
- [`_core.py`](_core.py): Implementation of the core entities in the IR, including `Model`, `Graph`, `Node`, `Value`, and others.
19+
- [`_enums.py`](_enums.py): Definition of the type enums that correspond to the `DataType` and `AttributeType` in `onnx.proto`.
20+
- [`_name_authority.py`](_name_authority.py): The authority for giving names to entities in the graph, used internally.
21+
- [`_linked_list.py`](_linked_list.py): The data structure as the node container in the graph that supports robust iteration and mutation. Internal.
22+
- [`_metadata.py`](_metadata.py): Metadata store for all entities in the IR.

0 commit comments

Comments
 (0)