We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 37036b3 commit 97daf75Copy full SHA for 97daf75
backends/xnnpack/test/tester/tester.py
@@ -496,14 +496,14 @@ def generate_random_inputs(self):
496
# the function to compute the derived dim.
497
dim_spec = dim_spec.root
498
fn = dim_spec.fn
499
- elif isinstance(dim_spec, torch.export.dynamic_shapes._Dim):
+ elif isinstance(dim_spec, torch.export.dynamic_shapes.Dim):
500
# Not derived dim so fn is just itself
501
def fn(x):
502
return x
503
504
else:
505
raise RuntimeError(
506
- f"Expected Dynamic Dims to be of type _DerivedDim or _Dim but got {type(dim_spec)}"
+ f"Expected Dynamic Dims to be of type _DerivedDim or Dim but got {type(dim_spec)}"
507
)
508
dim_name = dim_spec.__name__
509
if dim_name not in dim_name_to_size:
0 commit comments