search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 480 forks source link

[torch_xla2] Fix nn.functional.conv2d and conv3d #8048

Closed dvhg closed 1 month ago

dvhg commented 1 month ago

Replicate PyTorch behavior when using single number padding. Ref: https://pytorch.org/docs/stable/generated/torch.nn.functional.conv2d.html#torch.nn.functional.conv2d