Home
last modified time | relevance | path

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

/external/pytorch/torch/nn/modules/
Dpixelshuffle.py10 class PixelShuffle(Module): class
/external/executorch/backends/qualcomm/tests/
Dmodels.py704 class PixelShuffle(torch.nn.Module): class
/external/executorch/backends/apple/mps/test/
Dtest_mps.py201 class PixelShuffle(torch.nn.Module): class
/external/pytorch/test/cpp/api/
Dfunctional.cpp1751 TEST_F(FunctionalTest, PixelShuffle) { in TEST_F() argument
Dmodules.cpp2917 TEST_F(ModulesTest, PixelShuffle) { in TEST_F() argument
/external/pytorch/test/onnx/
Dtest_pytorch_onnx_onnxruntime.py7070 class PixelShuffle(torch.nn.Module): class