Home
last modified time | relevance | path

Searched refs:ChannelDimension (Results 1 – 10 of 10) sorted by relevance

/external/ruy/ruy/
Dperchannel_buffers_reallocation_test.cc71 ChannelDimension channel_dimension, in TestPerChannelBuffersReallocation()
82 test_advisory(true, ChannelDimension::kRow, 1); in TestPerChannelBuffersReallocation()
83 test_advisory(true, ChannelDimension::kRow, 2); in TestPerChannelBuffersReallocation()
84 test_advisory(true, ChannelDimension::kRow, 4); in TestPerChannelBuffersReallocation()
85 test_advisory(true, ChannelDimension::kRow, 8); in TestPerChannelBuffersReallocation()
86 test_advisory(false, ChannelDimension::kRow, 16); in TestPerChannelBuffersReallocation()
87 test_advisory(false, ChannelDimension::kRow, 32); in TestPerChannelBuffersReallocation()
88 test_advisory(false, ChannelDimension::kRow, 64); in TestPerChannelBuffersReallocation()
91 test_advisory(true, ChannelDimension::kCol, 1); in TestPerChannelBuffersReallocation()
92 test_advisory(true, ChannelDimension::kCol, 2); in TestPerChannelBuffersReallocation()
[all …]
Dmul_params.h31 enum class ChannelDimension : std::int8_t { enum
165 ChannelDimension channel_dimension() const { in channel_dimension()
168 void set_channel_dimension(ChannelDimension value) { in set_channel_dimension()
224 ChannelDimension channel_dimension = ChannelDimension::kRow;
264 ChannelDimension channel_dimension = ChannelDimension::kRow;
279 ChannelDimension channel_dimension = ChannelDimension::kRow;
Dcreate_trmul_params.h284 mul_params->channel_dimension() == ChannelDimension::kRow ? Side::kLhs
328 mul_params->channel_dimension() == ChannelDimension::kRow ? Side::kLhs
358 mul_params->channel_dimension() == ChannelDimension::kRow ? Side::kLhs
387 ChannelDimension channel_dimension, Ctx* ctx,
409 ChannelDimension channel_dimension, Ctx* ctx, TrMulParams* params) {
446 inline ChannelDimension Transpose(ChannelDimension channel_dimension) {
447 return channel_dimension == ChannelDimension::kCol ? ChannelDimension::kRow
448 : ChannelDimension::kCol;
470 ChannelDimension channel_dimension = mul_params.channel_dimension();
Dmul_params_test.cc40 EXPECT_EQ(mul_params.channel_dimension(), ChannelDimension::kRow); in TEST()
46 mul_params.set_channel_dimension(ChannelDimension::kCol); in TEST()
51 EXPECT_EQ(mul_params.channel_dimension(), ChannelDimension::kCol); in TEST()
Dkernel_arm.h68 mul_params.channel_dimension() == ChannelDimension::kRow) {
96 mul_params.channel_dimension() == ChannelDimension::kRow) {
120 mul_params.channel_dimension() == ChannelDimension::kRow) {
Dreference_mul.h41 mul_params.channel_dimension() == ChannelDimension::kRow ? i : j; in ReferenceMul()
Dkernel_x86.h55 mul_params.channel_dimension() == ChannelDimension::kRow) {
80 mul_params.channel_dimension() == ChannelDimension::kRow) {
106 mul_params.channel_dimension() == ChannelDimension::kRow) {
131 mul_params.channel_dimension() == ChannelDimension::kRow) {
156 mul_params.channel_dimension() == ChannelDimension::kRow) {
181 mul_params.channel_dimension() == ChannelDimension::kRow) {
Dkernel_common.h163 if (mul_params.channel_dimension() == ChannelDimension::kCol) {
252 if (mul_params.channel_dimension() == ChannelDimension::kCol) {
Dkernel.h220 mul_params.channel_dimension() == ChannelDimension::kRow ? i : j;
Dtest.h1625 ChannelDimension channel_dimension =
1627 ? ChannelDimension::kRow
1628 : ChannelDimension::kCol;
1630 const int num_channels = channel_dimension == ChannelDimension::kRow
1881 mul_params.channel_dimension() == ChannelDimension::kRow;
1884 mul_params.channel_dimension() == ChannelDimension::kRow;