conv2d() — pytorch Function Reference

Architecture documentation for the conv2d() function in generate_test_torchscripts.py from the pytorch codebase.

Function python CoreTensor Dispatcher calls 1

Entity Profile

CoreTensor→ Dispatcher→ conv2d() — pytorch Function Reference

Dependency Diagram

graph TD
  c96a71b2_2511_a9f3_a220_200ca87751a3["conv2d()"]
  c800a173_9551_2901_c091_d157ae22ae5a["contiguous()"]
  c96a71b2_2511_a9f3_a220_200ca87751a3 -->|calls| c800a173_9551_2901_c091_d157ae22ae5a
  style c96a71b2_2511_a9f3_a220_200ca87751a3 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/generate_test_torchscripts.py lines 124–130

    def conv2d(self, x: Tensor, w: Tensor, toChannelsLast: bool) -> Tensor:
        r = torch.nn.functional.conv2d(x, w)
        if toChannelsLast:
            r = r.contiguous(memory_format=torch.channels_last)
        else:
            r = r.contiguous()
        return r