Home
last modified time | relevance | path

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

/external/ruy/ruy/
Dmul_params.h165 ChannelDimension channel_dimension() const { in channel_dimension() function
166 return storage_.channel_dimension; in channel_dimension()
169 storage_.channel_dimension = value; in set_channel_dimension()
224 ChannelDimension channel_dimension = ChannelDimension::kRow; member
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,
396 dst_mul_params->set_channel_dimension(channel_dimension);
409 ChannelDimension channel_dimension, Ctx* ctx, TrMulParams* params) {
441 FinalizeMulParams(mul_params, channel_dimension, ctx, params);
446 inline ChannelDimension Transpose(ChannelDimension channel_dimension) {
447 return channel_dimension == ChannelDimension::kCol ? ChannelDimension::kRow
470 ChannelDimension channel_dimension = mul_params.channel_dimension();
[all …]
Dmul_params_test.cc40 EXPECT_EQ(mul_params.channel_dimension(), ChannelDimension::kRow); in TEST()
51 EXPECT_EQ(mul_params.channel_dimension(), ChannelDimension::kCol); in TEST()
Dperchannel_buffers_reallocation_test.cc71 ChannelDimension channel_dimension, in TestPerChannelBuffersReallocation()
73 mul_params.set_channel_dimension(channel_dimension); in TestPerChannelBuffersReallocation()
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 =
1629 test_set->mul_params.set_channel_dimension(channel_dimension);
1630 const int num_channels = channel_dimension == ChannelDimension::kRow
1881 mul_params.channel_dimension() == ChannelDimension::kRow;
1884 mul_params.channel_dimension() == ChannelDimension::kRow;