(shape: torch.Size)
| 131 | |
| 132 | |
| 133 | def determine_tensor_dynanism(shape: torch.Size) -> TensorShapeDynamism: |
| 134 | if all(isinstance(s, int) for s in shape): |
| 135 | return TensorShapeDynamism.STATIC |
| 136 | else: |
| 137 | try: |
| 138 | _ = eval_shape(shape) |
| 139 | return TensorShapeDynamism.DYNAMIC_BOUND |
| 140 | except torch.fx.experimental.symbolic_shapes.GuardOnDataDependentSymNode: |
| 141 | return TensorShapeDynamism.DYNAMIC_UNBOUND |
| 142 | |
| 143 | |
| 144 | ALIGNMENT = 16 |
no test coverage detected