Skip to content

Commit df8fc61

Browse files
authored
[jit] Remove @torch.jit.export (#10824)
Summary: #10795 As part of the process of deprecating torchscript. Test Plan: Reviewers: Subscribers: Tasks: Tags: ### Summary [PLEASE REMOVE] See [CONTRIBUTING.md's Pull Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests) for ExecuTorch PR guidelines. [PLEASE REMOVE] If this PR closes an issue, please add a `Fixes #<issue-id>` line. [PLEASE REMOVE] If this PR introduces a fix or feature that should be the upcoming release notes, please add a "Release notes: <area>" label. For a list of available release notes labels, check out [CONTRIBUTING.md's Pull Requests](https://github.com/pytorch/executorch/blob/main/CONTRIBUTING.md#pull-requests). ### Test plan [PLEASE REMOVE] How did you test this PR? Please write down any manual commands you used and note down tests that you have written if applicable.
1 parent 1da5168 commit df8fc61

File tree

4 files changed

+0
-5
lines changed

4 files changed

+0
-5
lines changed

backends/qualcomm/quantizer/observers/per_channel_param_observer.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,5 @@ def forward(self, x_orig):
109109
self.quant_max,
110110
)
111111

112-
@torch.jit.export
113112
def calculate_qparams(self):
114113
return self._calculate_qparams(self.min_val, self.max_val)

examples/cadence/models/rnnt_encoder.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,6 @@ def forward(
103103
layer_norm_out = self.layer_norm(output_linear_out)
104104
return layer_norm_out, transformer_lengths
105105

106-
@torch.jit.export
107106
def infer(
108107
self,
109108
input: torch.Tensor,

examples/models/efficient_sam/efficient_sam_core/efficient_sam.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,6 @@ def __init__(
5959
"pixel_std", torch.Tensor(pixel_std).view(1, 3, 1, 1), False
6060
)
6161

62-
@torch.jit.export
6362
def predict_masks(
6463
self,
6564
image_embeddings: torch.Tensor,
@@ -174,7 +173,6 @@ def get_rescaled_pts(
174173
dim=-1,
175174
)
176175

177-
@torch.jit.export
178176
def get_image_embeddings(self, batched_images) -> torch.Tensor:
179177
"""
180178
Predicts masks end-to-end from provided images and prompts.

examples/models/efficient_sam/efficient_sam_core/efficient_sam_encoder.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,6 @@ def forward(self, x):
142142
return x
143143

144144

145-
@torch.jit.export
146145
def get_abs_pos(
147146
abs_pos: torch.Tensor, has_cls_token: bool, hw: List[int]
148147
) -> torch.Tensor:

0 commit comments

Comments
 (0)