Sum reduction.
(x: torch.Tensor, dim: int = -1)
| 70 | |
| 71 | # Parallel Reductions |
| 72 | def reduce_sum_ref(x: torch.Tensor, dim: int = -1) -> torch.Tensor: |
| 73 | """Sum reduction.""" |
| 74 | return x.sum(dim=dim) |
| 75 | |
| 76 | def reduce_max_ref(x: torch.Tensor, dim: int = -1) -> torch.Tensor: |
| 77 | """Max reduction.""" |
nothing calls this directly
no outgoing calls
no test coverage detected