Output buffer (by shape): materialized empty where outputs are written in place (torch), omitted where they're returned functionally (jax).
| 42 | |
| 43 | |
| 44 | class OutTensor: |
| 45 | """Output buffer (by shape): materialized empty where outputs are written in |
| 46 | place (torch), omitted where they're returned functionally (jax).""" |
| 47 | |
| 48 | def __init__(self, shape, dtype="float32"): |
| 49 | self.shape = tuple(shape) |
| 50 | self.dtype = dtype |
| 51 | |
| 52 | |
| 53 | class ChallengeBase(ABC): |
no outgoing calls
no test coverage detected