Home
last modified time | relevance | path

Searched refs:group_channels (Results 1 – 6 of 6) sorted by relevance

/external/XNNPACK/test/
Dchannel-shuffle-nc.cc15 for (size_t group_channels = 1; group_channels < 100; group_channels += 15) { in TEST() local
19 .group_channels(group_channels) in TEST()
26 for (size_t group_channels = 1; group_channels < 100; group_channels += 15) { in TEST() local
30 .group_channels(group_channels) in TEST()
37 for (size_t group_channels = 1; group_channels < 100; group_channels += 15) { in TEST() local
41 .group_channels(group_channels) in TEST()
49 for (size_t group_channels = 1; group_channels < 100; group_channels += 15) { in TEST() local
53 .group_channels(group_channels) in TEST()
61 for (size_t group_channels = 1; group_channels < 100; group_channels += 15) { in TEST() local
65 .group_channels(group_channels) in TEST()
[all …]
Dchannel-shuffle-operator-tester.h36 inline ChannelShuffleOperatorTester& group_channels(size_t group_channels) { in group_channels() function
37 assert(group_channels != 0); in group_channels()
38 this->group_channels_ = group_channels; in group_channels()
42 inline size_t group_channels() const { in group_channels() function
47 return groups() * group_channels(); in channels()
116 groups(), group_channels(), in TestX8()
137 for (size_t c = 0; c < group_channels(); c++) { in TestX8()
138 ASSERT_EQ(uint32_t(input[i * input_stride() + g * group_channels() + c]), in TestX8()
164 groups(), group_channels(), in TestX32()
185 for (size_t c = 0; c < group_channels(); c++) { in TestX32()
[all …]
/external/XNNPACK/src/
Dchannel-shuffle-nc.c24 size_t group_channels, in create_channel_shuffle_nc() argument
47 if (group_channels == 0) { in create_channel_shuffle_nc()
50 group_channels); in create_channel_shuffle_nc()
54 const size_t channels = groups * group_channels; in create_channel_shuffle_nc()
59 input_stride, groups, group_channels); in create_channel_shuffle_nc()
67 output_stride, groups, group_channels); in create_channel_shuffle_nc()
80 channel_shuffle_op->group_channels = group_channels; in create_channel_shuffle_nc()
100 size_t group_channels, in xnn_create_channel_shuffle_nc_x8() argument
108 group_channels, in xnn_create_channel_shuffle_nc_x8()
118 size_t group_channels, in xnn_create_channel_shuffle_nc_x32() argument
[all …]
/external/XNNPACK/bench/
Dchannel-shuffle.cc24 const size_t group_channels = static_cast<size_t>(state.range(2)); in channel_shuffle_x8() local
30 …d::vector<uint8_t> input(XNN_EXTRA_BYTES / sizeof(uint8_t) + batch_size * groups * group_channels); in channel_shuffle_x8()
31 std::vector<uint8_t> output(batch_size * groups * group_channels); in channel_shuffle_x8()
42 groups, group_channels, in channel_shuffle_x8()
43 groups * group_channels /* input stride */, in channel_shuffle_x8()
44 groups * group_channels /* output stride */, in channel_shuffle_x8()
77 const size_t elements_per_iteration = batch_size * groups * group_channels; in channel_shuffle_x8()
89 const size_t group_channels = static_cast<size_t>(state.range(2)); in channel_shuffle_x32() local
95 std::vector<float> input(XNN_EXTRA_BYTES / sizeof(float) + batch_size * groups * group_channels); in channel_shuffle_x32()
96 std::vector<float> output(batch_size * groups * group_channels); in channel_shuffle_x32()
[all …]
/external/XNNPACK/src/xnnpack/
Doperator.h186 size_t group_channels; member
/external/XNNPACK/include/
Dxnnpack.h953 size_t group_channels,
1283 size_t group_channels,