Skip to content

Commit df6d664

Browse files
committed
Address comments
1 parent 9a0b1ed commit df6d664

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

test/stablehlo/test_unbounded_dynamism.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
import numpy as np
88
import torch
99
import torch_xla
10-
import torch_xla.core.xla_model as xm
1110
from torch.export import Dim, export
1211
from torch_xla.stablehlo import exported_program_to_stablehlo
1312

torch_xla/experimental/scan_layers.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ def scan_layers(layers: Iterable[torch.nn.Module],
4747
4848
Example:
4949
50-
>>> import torch_xla.core.xla_model as xm
5150
>>> import torch
5251
>>> import torch.nn as nn
5352
>>> from torch_xla.experimental.scan_layers import scan_layers

0 commit comments

Comments
 (0)