Home
last modified time | relevance | path

Searched refs:kRowMajor (Results 1 – 25 of 36) sorted by relevance

12

/external/ruy/ruy/
Dmatrix_test.cc32 layout.set_order(Order::kRowMajor); in TEST()
36 EXPECT_EQ(layout.order(), Order::kRowMajor); in TEST()
46 MakeSimpleLayout(321, 654, Order::kRowMajor, &layout); in TEST()
50 EXPECT_EQ(layout.order(), Order::kRowMajor); in TEST()
82 MakeSimpleLayout(12, 34, Order::kRowMajor, matrix.mutable_layout()); in TEST()
92 EXPECT_EQ(matrix.layout().order(), Order::kRowMajor); in TEST()
Dkernel_arm.h142 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
143 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
166 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
167 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 4>;
189 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
190 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
Dmat.h323 return layout.order == Order::kRowMajor; in IsRowMajor()
345 return layout.order == Order::kRowMajor; in IsRowMajor()
368 int col_stride = layout.order == Order::kRowMajor ? 1 : layout.stride; in Offset()
381 layout.order == Order::kRowMajor ? layout.kernel.rows : layout.stride; in Offset()
389 layout.kernel.order == Order::kRowMajor ? 1 : layout.kernel.rows; in Offset()
445 return order == Order::kColMajor ? Order::kRowMajor : Order::kColMajor; in Transpose()
Dpack_x86.h158 struct PackImpl<Path::kAvx, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
160 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
192 struct PackImpl<Path::kAvx2Fma, FixedKernelLayout<Order::kRowMajor, 1, 8>,
194 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
279 struct PackImpl<Path::kAvx512, FixedKernelLayout<Order::kRowMajor, 1, 16>,
284 using Layout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
316 Scalar, std::int8_t, std::int32_t, Order::kRowMajor> {
321 RUY_DCHECK_EQ(src_matrix.layout.order, Order::kRowMajor);
351 std::int8_t, std::int32_t, Order::kRowMajor> {
356 RUY_DCHECK_EQ(src_matrix.layout.order, Order::kRowMajor);
[all …]
Dbenchmark.cc67 test_set.lhs_order = orders[0] == 'R' ? Order::kRowMajor : Order::kColMajor; in Benchmark()
68 test_set.rhs_order = orders[1] == 'R' ? Order::kRowMajor : Order::kColMajor; in Benchmark()
69 test_set.dst_order = orders[2] == 'R' ? Order::kRowMajor : Order::kColMajor; in Benchmark()
Dtest.h516 return layout.order() == Order::kRowMajor;
647 Order lhs_order = Order::kRowMajor;
749 : Order::kRowMajor));
760 : Order::kRowMajor));
775 struct GemmlowpOrder<Order::kRowMajor> {
878 return (LhsOrder == Order::kRowMajor ? 4 : 0) +
879 (RhsOrder == Order::kRowMajor ? 2 : 0) +
880 (DstOrder == Order::kRowMajor ? 1 : 0);
897 EVALGEMMLOWP_CASE3(LHS, RHS, Order::kRowMajor)
900 EVALGEMMLOWP_CASE2(LHS, Order::kRowMajor)
[all …]
Dpack_common.h137 struct PackImpl<ThePath, FixedKernelLayout<Order::kRowMajor, 1, KernelCols>, \
138 float, float, float, Order::kRowMajor> \
Dprepacked_cache_test.cc210 mat.layout.order = Order::kRowMajor; in TEST()
222 mat.layout.kernel.order = Order::kRowMajor; in TEST()
240 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in TestCachePolicies()
281 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in TEST()
Dkernel.h153 using Lhs = FixedKernelLayout<Order::kRowMajor, 4, 4>;
154 using Rhs = FixedKernelLayout<Order::kRowMajor, 4, 4>;
168 using Rhs = FixedKernelLayout<Order::kRowMajor, 2, 8>;
Dmatrix.h29 enum class Order : std::uint8_t { kColMajor, kRowMajor }; enumerator
197 int col_stride = layout.order() == Order::kRowMajor ? 1 : layout.stride(); in Offset()
Dpack_arm.h363 struct PackImpl<Path::kNeon, FixedKernelLayout<Order::kRowMajor, 1, 8>, float,
439 struct PackImpl<Path::kNeon, FixedKernelLayout<Order::kRowMajor, 1, 4>, float,
499 Scalar, std::int8_t, std::int32_t, Order::kRowMajor> {
580 std::int8_t, std::int32_t, Order::kRowMajor> {
587 RUY_DCHECK_EQ(src_matrix.layout.order, Order::kRowMajor);
Dkernel_x86.h70 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
71 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 16>;
121 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
122 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
146 using LhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
147 using RhsLayout = FixedKernelLayout<Order::kRowMajor, 1, 8>;
Dpack.h149 Order::kRowMajor>::Run(tuning, src, &packed, start_col, end_col); in RunPack()
Dtrace.h382 return o == Order::kRowMajor ? "row-major" : "column-major"; in str()
391 layout.order() == Order::kRowMajor ? layout.cols() : layout.rows(); in str()
403 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str()
415 int inner_size = layout.order == Order::kRowMajor ? layout.cols : layout.rows; in str()
Dperchannel_buffers_reallocation_test.cc53 ruy::MakeSimpleLayout(kMatrixSize, kMatrixSize, ruy::Order::kRowMajor, in TestPerChannelBuffersReallocation()
/external/ruy/example/
Dexample.cc27 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulFloat()
52 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
79 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
110 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulInt8PerChannelQuantized()
135 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulInt8GetRawAccumulators()
Dparametrized_example.cc75 return ruy::Order::kRowMajor; in run()
89 int base_stride = order == ruy::Order::kRowMajor ? cols : rows; in run()
109 layout.order() == ruy::Order::kRowMajor ? layout.rows() : layout.cols(); in run()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dbatch_matmul.h81 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in BatchMatMul()
191 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in BatchMatMul()
335 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in BatchMatMul()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dfully_connected.h67 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in FullyConnected()
Dtranspose_conv.h65 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in TransposeConvV2()
Dconv.h100 lhs_params.order = cpu_backend_gemm::Order::kRowMajor; in ConvPerChannel()
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_params.h30 enum class Order { kColMajor, kRowMajor }; enumerator
Dcpu_backend_gemm.h132 if (lhs_params.order != Order::kRowMajor || in Gemm()
Dcpu_backend_gemm_ruy.h49 : ruy::Order::kRowMajor;
Dcpu_backend_gemm_test.cc393 return random_engine() % 2 ? cpu_backend_gemm::Order::kRowMajor in TestSomeGemm()
398 use_golden ? cpu_backend_gemm::Order::kRowMajor : random_order(); in TestSomeGemm()

12