Home
last modified time | relevance | path

Searched refs:argmaxpool (Results 1 – 9 of 9) sorted by relevance

/external/XNNPACK/test/
Dargmax-pooling-nhwc.cc32 …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 …]
Dargmaxpool-microkernel-tester.h147 …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/
Dgenerate-tests.sh34 tools/generate-argmaxpool-test.py --spec test/f32-argmaxpool.yaml --output test/f32-argmaxpool.cc
/external/XNNPACK/src/
Dinit.c336 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/
Dargmax-pooling-nhwc.c221 …st struct argmaxpool_parameters* ukernel = select_ukernel(pooling_size, xnn_params.f32.argmaxpool); in xnn_setup_argmax_pooling2d_nhwc_f32()
/external/XNNPACK/
DCMakeLists.txt252 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 …]
DBUILD.bazel126 "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 …]
DAndroid.bp110 "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/
Dparams.h2048 struct argmaxpool_parameters argmaxpool[XNN_MAX_F32_ARGMAXPOOL_UKERNELS]; member