Home
last modified time | relevance | path

Searched refs:MakeSimpleLayout (Results 1 – 9 of 9) sorted by relevance

/external/ruy/example/
Dexample.cc27 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulFloat()
30 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, rhs.mutable_layout()); in ExampleMulFloat()
33 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, dst.mutable_layout()); in ExampleMulFloat()
52 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
55 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, rhs.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
58 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, dst.mutable_layout()); in ExampleMulFloatWithBiasAddAndClamp()
79 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
83 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, rhs.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
87 ruy::MakeSimpleLayout(2, 2, ruy::Order::kColMajor, dst.mutable_layout()); in ExampleMulUint8AsymmetricQuantized()
110 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in ExampleMulInt8PerChannelQuantized()
[all …]
/external/ruy/ruy/
Dmatrix_test.cc39 TEST(MatrixTest, MakeSimpleLayout) { in TEST() argument
41 MakeSimpleLayout(123, 456, Order::kColMajor, &layout); in TEST()
46 MakeSimpleLayout(321, 654, Order::kRowMajor, &layout); in TEST()
82 MakeSimpleLayout(12, 34, Order::kRowMajor, matrix.mutable_layout()); in TEST()
Dperchannel_buffers_reallocation_test.cc53 ruy::MakeSimpleLayout(kMatrixSize, kMatrixSize, ruy::Order::kRowMajor, in TestPerChannelBuffersReallocation()
58 ruy::MakeSimpleLayout(kMatrixSize, kMatrixSize, ruy::Order::kColMajor, in TestPerChannelBuffersReallocation()
64 ruy::MakeSimpleLayout(kMatrixSize, kMatrixSize, ruy::Order::kColMajor, in TestPerChannelBuffersReallocation()
Dprepacked_cache_test.cc240 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in TestCachePolicies()
243 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, rhs.mutable_layout()); in TestCachePolicies()
246 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, dst.mutable_layout()); in TestCachePolicies()
281 ruy::MakeSimpleLayout(2, 2, ruy::Order::kRowMajor, lhs.mutable_layout()); in TEST()
284 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, rhs.mutable_layout()); in TEST()
287 ruy::MakeSimpleLayout(2, 1, ruy::Order::kColMajor, dst.mutable_layout()); in TEST()
Dmatrix.h169 inline void MakeSimpleLayout(int rows, int cols, Order order, Layout* layout) { in MakeSimpleLayout() function
/external/XNNPACK/bench/
Dqu8-gemm.cc236 ruy::MakeSimpleLayout(nc, kc, ruy::Order::kRowMajor, ruy_a.mutable_layout()); in RuyBenchmark()
239 ruy::MakeSimpleLayout(kc, mc, ruy::Order::kColMajor, ruy_b.mutable_layout()); in RuyBenchmark()
243 ruy::MakeSimpleLayout(nc, mc, ruy::Order::kColMajor, ruy_c.mutable_layout()); in RuyBenchmark()
Dqs8-gemm.cc223 ruy::MakeSimpleLayout(nc, kc, ruy::Order::kRowMajor, ruy_a.mutable_layout()); in RuyBenchmark()
226 ruy::MakeSimpleLayout(kc, mc, ruy::Order::kColMajor, ruy_b.mutable_layout()); in RuyBenchmark()
230 ruy::MakeSimpleLayout(nc, mc, ruy::Order::kColMajor, ruy_c.mutable_layout()); in RuyBenchmark()
Df32-gemm.cc292 ruy::MakeSimpleLayout(nc, kc, ruy::Order::kRowMajor, ruy_a.mutable_layout()); in RuyBenchmark()
294 ruy::MakeSimpleLayout(kc, mc, ruy::Order::kColMajor, ruy_b.mutable_layout()); in RuyBenchmark()
297 ruy::MakeSimpleLayout(nc, mc, ruy::Order::kColMajor, ruy_c.mutable_layout()); in RuyBenchmark()
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_gemm_ruy.h50 ruy::MakeSimpleLayout(params.rows, params.cols, ruy_order,