Skip to content

Commit e42c965

Browse files
committed
Recipe and Input class definitions with e2e export
Differential Revision: [D71946730](https://our.internmc.facebook.com/intern/diff/D71946730/) [ghstack-poisoned]
1 parent 1facfa9 commit e42c965

File tree

7 files changed

+550
-2
lines changed

7 files changed

+550
-2
lines changed

export/TARGETS

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
2+
3+
oncall("executorch")
4+
5+
python_library(
6+
name = "recipe",
7+
srcs = [
8+
"_recipe.py",
9+
],
10+
deps = [
11+
"//caffe2:torch",
12+
"//executorch/exir/backend:backend_api",
13+
"//executorch/exir:pass_manager",
14+
]
15+
)
16+
17+
python_library(
18+
name = "export",
19+
srcs = [
20+
"_export.py",
21+
],
22+
deps = [
23+
":recipe",
24+
]
25+
)
26+
27+
python_library(
28+
name = "lib",
29+
srcs = [
30+
"__init__.py",
31+
],
32+
deps = [
33+
":export",
34+
":recipe",
35+
],
36+
)

export/__init__.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
# Copyright (c) Meta Platforms, Inc. and affiliates.
2+
# All rights reserved.
3+
#
4+
# This source code is licensed under the BSD-style license found in the
5+
# LICENSE file in the root directory of this source tree.
6+
7+
"""
8+
ExecuTorch export module.
9+
10+
This module provides the tools and utilities for exporting PyTorch models
11+
to the ExecuTorch format, including configuration, quantization, and
12+
export management.
13+
"""
14+
15+
# pyre-strict
16+
17+
from ._export import ExportSession, export
18+
from ._recipe import ExportRecipe
19+
20+
__all__ = [
21+
"ExportRecipe",
22+
"ExportSession",
23+
"export",
24+
]

0 commit comments

Comments
 (0)