(xs, axis=0)
| 930 | ).sum() |
| 931 | |
| 932 | def reverse_sequence(xs, axis=0): |
| 933 | indices = mx.arange(xs.shape[axis] - 1, -1, -1) |
| 934 | return mx.take(xs, indices, axis=axis) |
| 935 | |
| 936 | def conv_manual_flip(kernel): |
| 937 | for ax in range(1, kernel.ndim - 1): |
nothing calls this directly
no outgoing calls
no test coverage detected