Lines Matching full:symint
17 def eval_expr(symint: Union[int, torch.SymInt]) -> Optional[int]: argument
19 Evaluate a symint to int. Returns None if symint's symoblic expr
22 if isinstance(symint, int):
23 return symint
24 node = symint.node
34 def eval_upper_bound(maybe_symint: Union[int, torch.SymInt]) -> int: argument
36 Evaluate a symint to its uppper bound value. Returns None if symint's symoblic expr's
69 def eval_shape(shape: Iterable[Union[int, torch.SymInt]]): # pyre-ignore[3] argument
80 def eval_shape_upper_bound(shape: Iterable[Union[int, torch.SymInt]]) -> List[int]: argument
88 shape: Iterable[Union[int, torch.SymInt]] argument
92 if not isinstance(sz, torch.SymInt):