Searched defs:KernelArgType (Results 1 – 3 of 3) sorted by relevance
26 def signature_of(arg: KernelArgType, *, size_dtype: str) -> str:68 signature: List[KernelArgType],109 args: List[KernelArgType],116 def is_aligned(x: KernelArgType, alignment: int, include_tensor: bool) -> bool:
94 KernelArgType = Union[WorkspaceArg, TensorArg, SizeArg] variable
48 pub enum KernelArgType { enum59 impl KernelArgType { impl