conv2d() — pytorch Function Reference

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

Function python AndroidRuntime LifeCycleTesting calls 1

Entity Profile

AndroidRuntime→ LifeCycleTesting→ conv2d() — pytorch Function Reference

Dependency Diagram

graph TD
  c218e441_44bb_1272_16f4_608fe92f50dd["conv2d()"]
  0b7e2e4a_6531_1537_a6f4_db8a7938bd78["contiguous()"]
  c218e441_44bb_1272_16f4_608fe92f50dd -->|calls| 0b7e2e4a_6531_1537_a6f4_db8a7938bd78
  style c218e441_44bb_1272_16f4_608fe92f50dd 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