conv3d() — pytorch Function Reference

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

Function python AndroidRuntime LifeCycleTesting calls 1

Entity Profile

AndroidRuntime→ LifeCycleTesting→ conv3d() — pytorch Function Reference

Dependency Diagram

graph TD
  c7b1c2e9_7a80_5b06_2ade_87c265f88be7["conv3d()"]
  0b7e2e4a_6531_1537_a6f4_db8a7938bd78["contiguous()"]
  c7b1c2e9_7a80_5b06_2ade_87c265f88be7 -->|calls| 0b7e2e4a_6531_1537_a6f4_db8a7938bd78
  style c7b1c2e9_7a80_5b06_2ade_87c265f88be7 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

android/pytorch_android/generate_test_torchscripts.py lines 133–139

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