Searched refs:constraint_sizes (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/fx/experimental/ |
| H A D | symbolic_shapes.py | 1300 constraint_sizes: DimList[DimConstraint] = None variable in StatelessSymbolicContext
1311 if self.constraint_sizes is None:
2992 constraint_dims = symbolic_context.constraint_sizes
3087 constraint_sizes = [None] * dim
3108 constraint_sizes=constraint_sizes,
3113 constraint_sizes = symbolic_context.constraint_sizes
3129 assert len(constraint_sizes) == dim
3434 symbolic_context.constraint_sizes[source.idx] = None
3670 constraint_sizes=[None] * t.dim(),
3936 (source, t, context.constraint_sizes, context.constraint_strides)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/variables/ |
| H A D | builder.py | 2400 constraint_sizes=outer_context.constraint_sizes,
2412 constraint_sizes=[None] * e.dim(),
2430 constraint_sizes=[None] * e.dim(),
2562 constraint_sizes = []
2633 constraint_sizes.append(constraint_size)
2664 constraint_sizes=constraint_sizes,
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | non_strict_utils.py | 99 constraint_sizes=[None] * n_dims,
105 symbolic_context.constraint_sizes[i] = constraint.constraint_range
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | meta_utils.py | 910 constraint_sizes=[None] * t.ndim,
918 constraint_sizes=[None] * t.ndim,
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | eval_frame.py | 911 constraint_sizes=[None] * len(flat_args[i].shape),
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_fake_tensor.py | 648 dynamic_sizes=[DimDynamic.DYNAMIC], constraint_sizes=[None]
|
| H A D | test_dynamic_shapes.py | 180 dynamic_sizes=dynamic_dims, constraint_sizes=constraint_dims
|