| /external/XNNPACK/test/ |
| D | prelu-nc.cc | 14 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
24 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
35 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
46 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
48 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
56 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
58 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
67 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
69 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
78 …annels = 1; channels < xnn_params.f32.prelu.channel_tile * 10; channels += std::max<size_t>(1, xnn… in TEST()
[all …]
|
| D | prelu-microkernel-tester.h | 118 void Test(xnn_f32_prelu_ukernel_function prelu, Variant variant = Variant::Native) const {
169 prelu(rows(), channels() * sizeof(float),
|
| /external/XNNPACK/scripts/ |
| D | generate-f32-prelu.sh | 9 tools/xngen src/f32-prelu/scalar.c.in -D CHANNEL_TILE=1 -D ROW_TILE=2 -D WASM=0 -o src/f32-prelu/ge…
10 tools/xngen src/f32-prelu/scalar.c.in -D CHANNEL_TILE=4 -D ROW_TILE=2 -D WASM=0 -o src/f32-prelu/ge…
13 tools/xngen src/f32-prelu/scalar.c.in -D CHANNEL_TILE=1 -D ROW_TILE=2 -D WASM=1 -o src/f32-prelu/ge…
14 tools/xngen src/f32-prelu/scalar.c.in -D CHANNEL_TILE=4 -D ROW_TILE=2 -D WASM=1 -o src/f32-prelu/ge…
17 tools/xngen src/f32-prelu/neon.c.in -D CHANNEL_TILE=4 -D ROW_TILE=2 -o src/f32-prelu/gen/neon-2x4.c
18 tools/xngen src/f32-prelu/neon.c.in -D CHANNEL_TILE=8 -D ROW_TILE=2 -o src/f32-prelu/gen/neon-2x8.c
21 tools/xngen src/f32-prelu/psimd.c.in -D CHANNEL_TILE=4 -D ROW_TILE=2 -o src/f32-prelu/gen/psimd-2x4…
22 tools/xngen src/f32-prelu/psimd.c.in -D CHANNEL_TILE=8 -D ROW_TILE=2 -o src/f32-prelu/gen/psimd-2x8…
25 tools/xngen src/f32-prelu/sse.c.in -D CHANNEL_TILE=4 -D ROW_TILE=2 -D BLEND=0 -o src/f32-prelu/gen/…
26 tools/xngen src/f32-prelu/sse.c.in -D CHANNEL_TILE=8 -D ROW_TILE=2 -D BLEND=0 -o src/f32-prelu/gen/…
[all …]
|
| /external/XNNPACK/src/ |
| D | prelu-nc.c | 128 prelu_op->context.prelu = (struct prelu_context) { in xnn_setup_prelu_nc_f32()
135 .ukernel = xnn_params.f32.prelu.ukernel, in xnn_setup_prelu_nc_f32()
145 const uint32_t row_tile = xnn_params.f32.prelu.row_tile; in xnn_setup_prelu_nc_f32()
|
| D | init.c | 247 xnn_params.f32.prelu = (struct prelu_parameters) { in init()
545 xnn_params.f32.prelu = (struct prelu_parameters) { in init()
906 xnn_params.f32.prelu = (struct prelu_parameters) { in init()
1215 xnn_params.f32.prelu = (struct prelu_parameters) { in init()
1423 xnn_params.f32.prelu = (struct prelu_parameters) { in init()
|
| /external/tensorflow/tensorflow/lite/testing/op_tests/ |
| D | prelu.py | 51 prelu = tf.keras.layers.PReLU(shared_axes=parameters["shared_axes"])
52 out = prelu(input_tensor)
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | BUILD | 27 ":prelu",
472 name = "prelu",
473 srcs = ["prelu.cc"],
474 hdrs = ["prelu.h"],
494 ":prelu",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | BUILD | 424 name = "prelu",
425 srcs = ["prelu.cc"],
426 hdrs = ["prelu.h"],
447 ":prelu",
672 "prelu",
|
| /external/XNNPACK/ |
| D | CMakeLists.txt | 150 src/prelu-nc.c
226 src/f32-prelu/gen/scalar-2x1.c
227 src/f32-prelu/gen/scalar-2x4.c
395 src/f32-prelu/gen/psimd-2x4.c
396 src/f32-prelu/gen/psimd-2x8.c
523 src/f32-prelu/gen/neon-2x4.c
524 src/f32-prelu/gen/neon-2x8.c
935 src/f32-prelu/gen/sse2-2x4.c
936 src/f32-prelu/gen/sse2-2x8.c
981 src/f32-prelu/gen/sse41-2x4.c
[all …]
|
| D | BUILD.bazel | 69 "src/prelu-nc.c",
134 "src/f32-prelu/gen/scalar-2x1.c",
135 "src/f32-prelu/gen/scalar-2x4.c",
282 "src/f32-prelu/gen/wasm-2x1.c",
283 "src/f32-prelu/gen/wasm-2x4.c",
388 "src/f32-prelu/gen/psimd-2x4.c",
389 "src/f32-prelu/gen/psimd-2x8.c",
519 "src/f32-prelu/gen/neon-2x4.c",
520 "src/f32-prelu/gen/neon-2x8.c",
944 "src/f32-prelu/gen/sse2-2x4.c",
[all …]
|
| D | Android.bp | 32 "src/prelu-nc.c",
97 "src/f32-prelu/gen/scalar-2x1.c",
98 "src/f32-prelu/gen/scalar-2x4.c",
267 "src/f32-prelu/gen/psimd-2x4.c",
268 "src/f32-prelu/gen/psimd-2x8.c",
398 "src/f32-prelu/gen/neon-2x4.c",
399 "src/f32-prelu/gen/neon-2x8.c",
815 "src/f32-prelu/gen/sse2-2x4.c",
816 "src/f32-prelu/gen/sse2-2x8.c",
862 "src/f32-prelu/gen/sse41-2x4.c",
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | operator.h | 276 struct prelu_context prelu; member
|
| D | params.h | 1374 struct prelu_parameters prelu; member
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | BUILD | 41 "prelu.cc",
111 "prelu.cc",
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/selectors/ |
| D | BUILD | 115 "//tensorflow/lite/delegates/gpu/cl/kernels:prelu",
|
| /external/XNNPACK/src/f32-prelu/ |
| D | scalar.c.in | 14 #include <xnnpack/prelu.h>
|
| D | psimd.c.in | 15 #include <xnnpack/prelu.h>
|
| D | neon.c.in | 15 #include <xnnpack/prelu.h>
|
| D | sse.c.in | 18 #include <xnnpack/prelu.h>
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | BUILD | 956 name = "prelu",
957 srcs = ["prelu.cc"],
958 hdrs = ["prelu.h"],
985 ":prelu",
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | BUILD | 451 "reference/prelu.h",
515 "reference/prelu.h",
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples_lib.py | 114 from tensorflow.lite.testing.op_tests.prelu import make_prelu_tests
|
| /external/tensorflow/tensorflow/lite/micro/tools/make/ |
| D | Makefile | 146 tensorflow/lite/kernels/internal/reference/prelu.h \
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | ops.mlir | 1418 …%0 = "tfl.prelu"(%arg0, %arg1) : (tensor<10x10x10x10xf32>, tensor<1x1x10xf32>) -> tensor<10x10x10x…
1426 %0 = "tfl.prelu"(%arg0, %arg1) : (tensor<1x2x3x4xf32>, tensor<2x3x4xf32>) -> tensor<1x2x3x5xf32>
1434 …%0 = "tfl.prelu"(%arg0, %arg1) : (tensor<7x3x2x14xf32>, tensor<2x7x3x2x14xf32>) -> tensor<7x3x2x14…
1442 …%0 = "tfl.prelu"(%arg0, %arg1) : (tensor<15x14x2x14xf32>, tensor<1x1x3xf32>) -> tensor<15x14x2x14x…
1449 …%0 = "tfl.prelu"(%arg0, %arg1) : (tensor<16x20x20x13xf32>, tensor<20x20x13xf32>) -> tensor<16x20x2…
1456 …%0 = "tfl.prelu"(%arg0, %arg1) : (tensor<19x7x12x14xf32>, tensor<1x1x14xf32>) -> tensor<19x7x12x14…
1463 %0 = "tfl.prelu"(%arg0, %arg1) : (tensor<7x8x9x10xf32>, tensor<1x1x1xf32>) -> tensor<7x8x9x10xf32>
1470 …%0 = "tfl.prelu"(%arg0, %arg1) : (tensor<1x96x96x16x!quant.uniform<u8:f32, 0.00784:128>>, tensor<1…
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 320 "prelu",
|