Skip to content

Commit 4b723c3

Browse files
AdrianLundellStrycekSimon
authored andcommitted
Arm backend: Update backends-arm-ethos-u.md. (pytorch#14255)
Update the doc to be more focused on detailed API documentation. Note that this documents the change introduced in pytorch#14191. --------- Signed-off-by: Adrian Lundell <[email protected]>
1 parent 1ecc5fc commit 4b723c3

File tree

2 files changed

+185
-97
lines changed

2 files changed

+185
-97
lines changed

backends/arm/ethosu/partitioner.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,14 @@
1515

1616
@final
1717
class EthosUPartitioner(TOSAPartitioner):
18+
"""
19+
Partitions subgraphs supported by the Arm Ethos-U backend.
20+
21+
Attributes:
22+
compile_spec: List of CompileSpec objects for Ethos-U backend.
23+
additional_checks: Optional sequence of additional operator support checks.
24+
"""
25+
1826
def __init__(
1927
self,
2028
compile_spec: EthosUCompileSpec,

0 commit comments

Comments
 (0)