Searched refs:kCol (Results 1 – 7 of 7) sorted by relevance
| /external/ruy/ruy/ |
| D | perchannel_buffers_reallocation_test.cc | 91 test_advisory(true, ChannelDimension::kCol, 1); in TestPerChannelBuffersReallocation()
92 test_advisory(true, ChannelDimension::kCol, 2); in TestPerChannelBuffersReallocation()
93 test_advisory(true, ChannelDimension::kCol, 4); in TestPerChannelBuffersReallocation()
94 test_advisory(false, ChannelDimension::kCol, 8); in TestPerChannelBuffersReallocation()
95 test_advisory(false, ChannelDimension::kCol, 16); in TestPerChannelBuffersReallocation()
96 test_advisory(false, ChannelDimension::kCol, 32); in TestPerChannelBuffersReallocation()
97 test_advisory(false, ChannelDimension::kCol, 64); in TestPerChannelBuffersReallocation()
|
| D | mul_params_test.cc | 46 mul_params.set_channel_dimension(ChannelDimension::kCol); in TEST()
51 EXPECT_EQ(mul_params.channel_dimension(), ChannelDimension::kCol); in TEST()
|
| D | kernel_common.h | 163 if (mul_params.channel_dimension() == ChannelDimension::kCol) {
252 if (mul_params.channel_dimension() == ChannelDimension::kCol) {
|
| D | mul_params.h | 35 kCol enumerator
|
| D | create_trmul_params.h | 447 return channel_dimension == ChannelDimension::kCol ? ChannelDimension::kRow
448 : ChannelDimension::kCol;
|
| D | test.h | 1628 : ChannelDimension::kCol;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | tensor_utils_test.cc | 346 constexpr int kCol = 4; in TEST() local
348 static float matrix[kRow * kCol] = {1.0, 2.0, 3.0, 4.0, // in TEST()
351 static float vector[kCol * kBatch] = {1.0, -1.0, 1.0, -1.0, // in TEST()
355 MatrixBatchVectorMultiplyAccumulate(matrix, kRow, kCol, vector, kBatch, in TEST()
1482 const int kCol = 48; in TEST() local
1485 float matrix[kRow * kCol] = { in TEST()
1530 float vector[kBatch * kCol] = { in TEST()
1545 MatrixBatchVectorMultiplyAccumulate(matrix, kRow, kCol, vector, kBatch, in TEST()
1555 matrix_values, ledger, kRow, kCol, vector, kBatch, sparse_output.data()); in TEST()
1565 const int kCol = 48; in TEST() local
[all …]
|