(x, value=0.0)
| 731 | |
| 732 | def test_vmap_pad(self): |
| 733 | def pad2d(x, value=0.0): |
| 734 | return mx.pad(x, ((1, 2), (0, 1)), constant_values=value) |
| 735 | |
| 736 | x = mx.arange(24, dtype=mx.float32).reshape(2, 3, 4) |
| 737 |
nothing calls this directly
no outgoing calls
no test coverage detected