Home
last modified time | relevance | path

Searched refs:row_tile (Results 1 – 7 of 7) sorted by relevance

/external/XNNPACK/test/
Dprelu-nc.cc48 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
58 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
69 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
80 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
92 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
103 .batch_size(xnn_params.f32.prelu.row_tile) in TEST()
114 .batch_size(3 * xnn_params.f32.prelu.row_tile + 1) in TEST()
124 .batch_size(3 * xnn_params.f32.prelu.row_tile + 1) in TEST()
135 .batch_size(3 * xnn_params.f32.prelu.row_tile + 1) in TEST()
146 .batch_size(3 * xnn_params.f32.prelu.row_tile + 1) in TEST()
/external/XNNPACK/tools/
Dgenerate-prelu-test.py32 row_tile = int(match.group(3))
36 return row_tile, channel_tile, arch, isa
200 def generate_test_cases(ukernel, row_tile, channel_tile, isa): argument
224 "ROW_TILE": row_tile,
261 row_tile, channel_tile, arch, isa = split_ukernel_name(name)
266 test_case = generate_test_cases(name, row_tile, channel_tile, isa)
Dgenerate-vmulcaddc-test.py34 row_tile = int(match.group(4))
37 return channel_tile, row_tile, arch, isa
208 def generate_test_cases(ukernel, channel_tile, row_tile, isa): argument
233 "ROW_TILE": row_tile,
269 channel_tile, row_tile, arch, isa = split_ukernel_name(name)
274 test_case = generate_test_cases(name, channel_tile, row_tile, isa)
/external/XNNPACK/src/
Dprelu-nc.c145 const uint32_t row_tile = xnn_params.f32.prelu.row_tile; in xnn_setup_prelu_nc_f32() local
146 … batch_tile = min(batch_tile, divide_round_up(batch_tile, max_batch_tile * row_tile) * row_tile); in xnn_setup_prelu_nc_f32()
Dinit.c249 .row_tile = 2, in init()
293 .row_tile = 2, in init()
547 .row_tile = 2, in init()
591 .row_tile = 2, in init()
908 .row_tile = 2, in init()
1028 .row_tile = 2, in init()
1217 .row_tile = 2, in init()
1261 .row_tile = 2, in init()
1425 .row_tile = 4, in init()
1469 .row_tile = 2, in init()
Dconvolution-nhwc.c574 .mr = xnn_params.f32.vmulcaddc.row_tile, in xnn_create_convolution2d_nhwc_f32()
/external/XNNPACK/src/xnnpack/
Dparams.h1321 uint16_t row_tile; member
1333 uint8_t row_tile; member