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.h37 inline ChannelShuffleOperatorTester& group_channels(size_t group_channels) { in group_channels() function
38 assert(group_channels != 0); in group_channels()
39 this->group_channels_ = group_channels; in group_channels()
43 inline size_t group_channels() const { in group_channels() function
48 return groups() * group_channels(); in channels()
117 groups(), group_channels(), in TestX8()
138 for (size_t c = 0; c < group_channels(); c++) { in TestX8()
139 ASSERT_EQ(uint32_t(input[i * input_stride() + g * group_channels() + c]), in TestX8()
165 groups(), group_channels(), in TestX32()
186 for (size_t c = 0; c < group_channels(); c++) { in TestX32()
[all …]
/external/XNNPACK/src/operators/
Dchannel-shuffle-nc.c24 size_t group_channels, in create_channel_shuffle_nc() argument
49 if (group_channels == 0) { in create_channel_shuffle_nc()
52 xnn_operator_type_to_string(operator_type), group_channels); in create_channel_shuffle_nc()
56 const size_t channels = groups * group_channels; in create_channel_shuffle_nc()
61 xnn_operator_type_to_string(operator_type), input_stride, groups, group_channels); in create_channel_shuffle_nc()
69 xnn_operator_type_to_string(operator_type), output_stride, groups, group_channels); in create_channel_shuffle_nc()
84 channel_shuffle_op->group_channels = group_channels; in create_channel_shuffle_nc()
103 size_t group_channels, in xnn_create_channel_shuffle_nc_x8() argument
111 group_channels, in xnn_create_channel_shuffle_nc_x8()
121 size_t group_channels, in xnn_create_channel_shuffle_nc_x32() argument
[all …]
/external/XNNPACK/bench/
Dchannel-shuffle.cc25 const size_t group_channels = static_cast<size_t>(state.range(2)); in channel_shuffle_x8() local
31 …d::vector<uint8_t> input(XNN_EXTRA_BYTES / sizeof(uint8_t) + batch_size * groups * group_channels); in channel_shuffle_x8()
32 std::vector<uint8_t> output(batch_size * groups * group_channels); in channel_shuffle_x8()
43 groups, group_channels, in channel_shuffle_x8()
44 groups * group_channels /* input stride */, in channel_shuffle_x8()
45 groups * group_channels /* output stride */, in channel_shuffle_x8()
81 const size_t elements_per_iteration = batch_size * groups * group_channels; in channel_shuffle_x8()
93 const size_t group_channels = static_cast<size_t>(state.range(2)); in channel_shuffle_x32() local
99 std::vector<float> input(XNN_EXTRA_BYTES / sizeof(float) + batch_size * groups * group_channels); in channel_shuffle_x32()
100 std::vector<float> output(batch_size * groups * group_channels); in channel_shuffle_x32()
[all …]
/external/XNNPACK/src/xnnpack/
Doperator.h205 size_t group_channels; member
/external/XNNPACK/include/
Dxnnpack.h1702 size_t group_channels,
2257 size_t group_channels,