(self, node, method="__call__")
| 429 | """ |
| 430 | |
| 431 | def __init__(self, node, method="__call__"): |
| 432 | super().__init__() |
| 433 | if isinstance(node, type): |
| 434 | assert issubclass(node, Tensor) |
| 435 | cls = Parameter if issubclass(node, Parameter) else Tensor |
| 436 | |
| 437 | self.inputs = [] |
| 438 | self.const_val = [(0, cls)] |
| 439 | else: |
| 440 | assert isinstance(node, (TensorNode, ModuleNode)) |
| 441 | node.users.append(self) |
| 442 | self.inputs = [ |
| 443 | node, |
| 444 | ] |
| 445 | self.const_val = [] |
| 446 | self.arg_def = tree_flatten(((node,), {}))[1] |
| 447 | self.method = method |
| 448 | |
| 449 | @classmethod |
| 450 | def make(cls, *args, **kwargs): |
nothing calls this directly
no test coverage detected