Lines Matching full:symint
99 symint: bool = False,
104 elif str(t) == "SymInt":
105 if symint:
117 elem = valuetype_type(t.elem, binds=binds, mutable=mutable, symint=symint)
141 symint: bool = False,
148 symint=symint,
178 elif isinstance(t.elem, ListType) and str(t.elem.elem) == "SymInt":
179 if symint:
183 elem = argumenttype_type(t.elem, mutable=mutable, binds=binds, symint=symint)
192 if str(t.elem) == "SymInt":
194 if symint:
199 if symint:
216 elem = argumenttype_type(t.elem, mutable=mutable, binds=binds, symint=symint)
223 def argument_type(a: Argument, *, binds: ArgName, symint: bool = False) -> NamedCType:
224 return argumenttype_type(a.type, mutable=a.is_write, symint=symint, binds=binds)
232 def returntype_type(t: Type, *, mutable: bool, symint: bool = False) -> CType:
234 # NB: symint is ALWAYS respected for return types. So symint argument
236 r = valuetype_type(t, binds="__placeholder__", mutable=mutable, symint=True)
271 def return_type(r: Return, *, symint: bool = False) -> CType:
272 return returntype_type(r.type, mutable=r.is_write, symint=symint)
276 def returns_type(rs: Sequence[Return], *, symint: bool = False) -> CType:
280 return return_type(rs[0], symint=symint)
282 return TupleCType([return_type(r, symint=symint) for r in rs])
329 def default_expr(d: str, t: Type, *, symint: bool) -> str:
357 return default_expr(d, t.elem, symint=symint)
362 elif symint and d.isdigit() and str(t.elem) == "SymInt":
363 return f"c10::SymInt({d})"
380 symint: bool = False,
391 symint=symint,
403 default = default_expr(a.default, a.type, symint=symint)
406 nctype=argument_type(a, binds=binds, symint=symint),
450 symint: bool = False,
467 symint=symint,