Home
last modified time | relevance | path

Searched defs:op_callable (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/fx/
H A Ddecomposition_skip.py88 op_callable = torch._C._nn.upsample_bilinear2d # type: ignore[attr-defined] variable in UpsampleBilinear2DDecompSkip
126 op_callable = torch._C._nn.upsample_trilinear3d # type: ignore[attr-defined] variable in UpsampleTrilinear3DDecompSkip
164 op_callable = torch.instance_norm # type: ignore[attr-defined] variable in InstanceNormDecompSkip