(self, *lst:Tensor, fxn:Tensor|UOp, grad_fxn:Callable|None=None)
| 208 | param = UOp.param(slot, self.dtype, self.shape, self.device) |
| 209 | return Tensor(param) |
| 210 | def call(self, *lst:Tensor, fxn:Tensor|UOp, grad_fxn:Callable|None=None) -> Tensor: |
| 211 | fret = (fxn.uop if isinstance(fxn, Tensor) else fxn).call(*[t.uop for t in (self,)+lst], grad_fxn=grad_fxn) |
| 212 | return Tensor(fret.gettuple(0)) |
| 213 | |
| 214 | def custom_kernel(self, *lst:Tensor, fxn:Callable, grad_fxn:Callable|None=None) -> list[Tensor]: |
| 215 | """ |