Lines Matching defs:Tensor

34         def fn(x: Tensor):
49         def fn(x: Tensor):
65             def __init__(self, a: Tensor, b: Tensor):
72         def fn(x: Tensor):
89             def __init__(self, a: Tensor, b: Tensor):
101         def fn(x: Tensor):
120             def __init__(self, a: Tensor, b: Tensor):
131         def fn(x: Tensor):
149             def __init__(self, a: Tensor, b: Tensor):
163         def fn(x: Tensor):
182             def __init__(self, a: Tensor, b: Tensor):
199         def fn(x: Tensor):
220             def __init__(self, a: Tensor, b: Tensor):
232         def fn(x: Tensor) -> Await[Await[Tensor]]:
235         def main(x: Tensor) -> Tensor:
265         def delayed(x: Tensor) -> Tensor:
268         def main(x: Tensor) -> Tensor:
285         def delayed(x: Tensor) -> Tensor:
294         def delayed(x: Tensor) -> Tensor:
297         def main(x: Tensor) -> Await[Tensor]:
315         def gap(x: Tensor):
318         def delayed(x: Tensor) -> Tensor:
321         def main(x: Tensor, y: Tensor) -> Tensor:
333         def gap(x: Tensor):
336         def delayed(x: Tensor) -> Tuple[Tensor, List[Tensor]]:
340         def main(x: Tensor) -> Tensor:
363         def gap(x: Tensor):
366         def delayed(x: Tensor) -> Tensor:
369         def fn(aw: Await[Tensor]) -> Tensor:
372         def main(x: Tensor) -> Tensor: