Searched refs:argmaxpool (Results 1 – 9 of 9) sorted by relevance
/external/XNNPACK/test/ |
D | argmax-pooling-nhwc.cc | 32 …for (size_t pool_size = 2; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 48 …for (size_t pool_size = 3; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 70 …for (size_t pool_size = 3; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 89 …for (size_t pool_size = 2; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 105 …for (size_t pool_size = 2; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 127 …for (size_t pool_size = 2; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 146 …for (size_t pool_size = 2; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 172 …for (size_t pool_size = 2; pool_size <= FindMaxSinglePassPoolingSize(xnn_params.f32.argmaxpool); p… in TEST() 197 const auto multipass = FindMultiPassMicroKernel(xnn_params.f32.argmaxpool); in TEST() 214 const auto multipass = FindMultiPassMicroKernel(xnn_params.f32.argmaxpool); in TEST() [all …]
|
D | argmaxpool-microkernel-tester.h | 147 …void Test(xnn_f32_argmaxpool_unipass_ukernel_function argmaxpool, Variant variant = Variant::Nativ… 187 argmaxpool(output_pixels(), pooling_elements(), channels(), 214 …void Test(xnn_f32_argmaxpool_multipass_ukernel_function argmaxpool, Variant variant = Variant::Nat… 258 argmaxpool(output_pixels(), pooling_elements(), channels(),
|
/external/XNNPACK/scripts/ |
D | generate-tests.sh | 34 tools/generate-argmaxpool-test.py --spec test/f32-argmaxpool.yaml --output test/f32-argmaxpool.cc
|
/external/XNNPACK/src/ |
D | init.c | 336 xnn_params.f32.argmaxpool[0] = (struct argmaxpool_parameters) { in init() 340 xnn_params.f32.argmaxpool[1] = (struct argmaxpool_parameters) { in init() 344 xnn_params.f32.argmaxpool[2] = (struct argmaxpool_parameters) { in init() 624 xnn_params.f32.argmaxpool[0] = (struct argmaxpool_parameters) { in init() 628 xnn_params.f32.argmaxpool[1] = (struct argmaxpool_parameters) { in init() 632 xnn_params.f32.argmaxpool[2] = (struct argmaxpool_parameters) { in init() 1265 xnn_params.f32.argmaxpool[0] = (struct argmaxpool_parameters) { in init() 1269 xnn_params.f32.argmaxpool[1] = (struct argmaxpool_parameters) { in init() 1273 xnn_params.f32.argmaxpool[2] = (struct argmaxpool_parameters) { in init() 1744 xnn_params.f32.argmaxpool[0] = (struct argmaxpool_parameters) { in init() [all …]
|
/external/XNNPACK/src/operators/ |
D | argmax-pooling-nhwc.c | 221 …st struct argmaxpool_parameters* ukernel = select_ukernel(pooling_size, xnn_params.f32.argmaxpool); in xnn_setup_argmax_pooling2d_nhwc_f32()
|
/external/XNNPACK/ |
D | CMakeLists.txt | 252 src/f32-argmaxpool/4x-scalar-c1.c 253 src/f32-argmaxpool/9p8x-scalar-c1.c 254 src/f32-argmaxpool/9x-scalar-c1.c 665 src/f32-argmaxpool/4x-neon-c4.c 666 src/f32-argmaxpool/9p8x-neon-c4.c 667 src/f32-argmaxpool/9x-neon-c4.c 1706 src/f32-argmaxpool/4x-sse2-c4.c 1707 src/f32-argmaxpool/9p8x-sse2-c4.c 1708 src/f32-argmaxpool/9x-sse2-c4.c 3472 ADD_EXECUTABLE(f32-argmaxpool-test test/f32-argmaxpool.cc) [all …]
|
D | BUILD.bazel | 126 "src/f32-argmaxpool/4x-scalar-c1.c", 127 "src/f32-argmaxpool/9p8x-scalar-c1.c", 128 "src/f32-argmaxpool/9x-scalar-c1.c", 727 "src/f32-argmaxpool/4x-wasmsimd-c4.c", 728 "src/f32-argmaxpool/9p8x-wasmsimd-c4.c", 729 "src/f32-argmaxpool/9x-wasmsimd-c4.c", 1416 "src/f32-argmaxpool/4x-neon-c4.c", 1417 "src/f32-argmaxpool/9p8x-neon-c4.c", 1418 "src/f32-argmaxpool/9x-neon-c4.c", 2464 "src/f32-argmaxpool/4x-sse2-c4.c", [all …]
|
D | Android.bp | 110 "src/f32-argmaxpool/4x-scalar-c1.c", 111 "src/f32-argmaxpool/9p8x-scalar-c1.c", 112 "src/f32-argmaxpool/9x-scalar-c1.c", 525 "src/f32-argmaxpool/4x-neon-c4.c", 526 "src/f32-argmaxpool/9p8x-neon-c4.c", 527 "src/f32-argmaxpool/9x-neon-c4.c", 1573 "src/f32-argmaxpool/4x-sse2-c4.c", 1574 "src/f32-argmaxpool/9p8x-sse2-c4.c", 1575 "src/f32-argmaxpool/9x-sse2-c4.c",
|
/external/XNNPACK/src/xnnpack/ |
D | params.h | 2048 struct argmaxpool_parameters argmaxpool[XNN_MAX_F32_ARGMAXPOOL_UKERNELS]; member
|