Home
last modified time | relevance | path

Searched refs:constraint_strides (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_dynamo/variables/
H A Dbuilder.py2401 constraint_strides=outer_context.constraint_strides,
2413 constraint_strides=[None] * e.dim(),
2431 constraint_strides=[None] * e.dim(),
2563 constraint_strides = []
2634 constraint_strides.append(constraint_stride)
2665 constraint_strides=constraint_strides,
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsymbolic_shapes.py1301 constraint_strides: DimList[DimConstraint] = None variable in StatelessSymbolicContext
1313 if self.constraint_strides is None:
3088 constraint_strides = [None] * dim
3109 constraint_strides=constraint_strides,
3114 constraint_strides = symbolic_context.constraint_strides
3130 assert len(constraint_strides) == dim
3180 constraint_dim=constraint_strides[i],
3671 constraint_strides=[None] * t.dim()
3936 (source, t, context.constraint_sizes, context.constraint_strides)
3946 inner_context.constraint_strides
[all …]