Home
last modified time | relevance | path

Searched refs:unpool (Results 1 – 11 of 11) sorted by relevance

/external/XNNPACK/test/
Dunpool-microkernel-tester.h76 void Test(xnn_x32_unpool_ukernel_function unpool) const { in Test() argument
105 unpool(p(), c(), f(), x.data(), i.data(), indirect_y.data()); in Test()
/external/pytorch/test/nn/
Dtest_pooling.py417 unpool = nn.MaxUnpool2d(3, stride=2).cpu()
420 upout = unpool(out, ind)
638 unpool = torch.nn.MaxUnpool1d(2, stride=2).to(device)
642 unpool_out = unpool(output, indices)
649 unpool = torch.nn.MaxUnpool2d(2, stride=2).to(device)
652 unpool_out = unpool(output, indices)
659 unpool = torch.nn.MaxUnpool3d(2, stride=2).to(device)
663 unpool_out = unpool(output, indices)
766 unpool = module_class(module_size).to(device)
775 unpool(output, indices)
[all …]
/external/XNNPACK/src/operators/
Dunpooling-nhwc.c214 .ukernel = xnn_params.x32.unpool, in xnn_setup_unpooling2d_nhwc_x32()
/external/XNNPACK/src/xnnpack/
Dparams.h660 xnn_unpool_ukernel_function unpool; member
/external/pytorch/test/cpp/api/
Dmodules.cpp938 MaxUnpool1d unpool{MaxUnpool1dOptions(2).stride(2)}; in TEST_F() local
942 unpool(output, indices), in TEST_F()
949 unpool(output, indices, input.sizes().vec()), in TEST_F()
952 unpool(output, indices), in TEST_F()
988 MaxUnpool2d unpool{MaxUnpool2dOptions(2).stride(2)}; in TEST_F() local
994 unpool(output, indices), in TEST_F()
1000 unpool(output, indices, std::vector<int64_t>{1, 1, 5, 5}), in TEST_F()
1051 MaxUnpool3d unpool{MaxUnpool3dOptions(3).stride(2)}; in TEST_F() local
1054 auto unpooled_output = unpool(output, indices); in TEST_F()
/external/XNNPACK/src/
Dinit.c1289 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__neon; in init()
1864 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__scalar; in init()
3453 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__neon; in init()
5340 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__sse2; in init()
6283 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__wasmsimd; in init()
7021 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__scalar; in init()
7595 xnn_params.x32.unpool = (xnn_unpool_ukernel_function) xnn_x32_unpool_ukernel__scalar; in init()
/external/XNNPACK/
DCMakeLists.txt452 src/x32-unpool/scalar.c
602 src/x32-unpool/scalar.c
1438 src/x32-unpool/scalar.c
1643 src/x32-unpool/neon.c
2763 src/x32-unpool/neon.c
4152 src/x32-unpool/sse2.c
4517 src/x32-unpool/sse2.c
8842 ADD_EXECUTABLE(x32-unpool-test test/x32-unpool.cc $<TARGET_OBJECTS:all_microkernels>)
8843 TARGET_INCLUDE_DIRECTORIES(x32-unpool-test PRIVATE include src test)
8844 TARGET_LINK_LIBRARIES(x32-unpool-test PRIVATE cpuinfo fp16 pthreadpool gtest gtest_main)
[all …]
DBUILD.bazel311 "src/x32-unpool/scalar.c",
462 "src/x32-unpool/scalar.c",
613 "src/x32-unpool/scalar.c",
1450 "src/x32-unpool/scalar.c",
2727 "src/x32-unpool/wasmsimd.c",
3148 "src/x32-unpool/neon.c",
4269 "src/x32-unpool/neon.c",
5685 "src/x32-unpool/sse2.c",
6051 "src/x32-unpool/sse2.c",
8622 "src/xnnpack/unpool.h",
[all …]
DAndroid.bp270 "src/x32-unpool/scalar.c",
421 "src/x32-unpool/scalar.c",
572 "src/x32-unpool/scalar.c",
1409 "src/x32-unpool/scalar.c",
2686 "src/x32-unpool/wasmsimd.c",
3107 "src/x32-unpool/neon.c",
4228 "src/x32-unpool/neon.c",
5644 "src/x32-unpool/sse2.c",
6010 "src/x32-unpool/sse2.c",
/external/pytorch/third_party/
Dxnnpack_src_defs.bzl493 "XNNPACK/src/configs/unpool-config.c",
/external/pytorch/test/
Dtest_nn.py7376 def __init__(self, pool, unpool): argument
7379 self.unpool = unpool
7382 return self.unpool(*self.pool(input))