Home
last modified time | relevance | path

Searched refs:SidePair (Results 1 – 7 of 7) sorted by relevance

/external/ruy/ruy/
Dblock_map.h90 SidePair<int> dims;
94 SidePair<int> rectangularness_log2;
97 SidePair<int> kernel_dims;
99 SidePair<int> small_block_dims;
103 SidePair<int> large_blocks;
125 SidePair<int>* block);
137 void GetBlockMatrixCoords(const BlockMap& block_map, const SidePair<int>& block,
138 SidePair<int>* start, SidePair<int>* end);
Dtrmul_params.h29 using RunKernelFn = void(Tuning, const SidePair<PEMat>&, const void*,
30 const SidePair<int>&, const SidePair<int>&, EMat*);
65 void RunKernel(Tuning tuning, const SidePair<int>& start, in RunKernel()
66 const SidePair<int>& end) { in RunKernel()
75 SidePair<RunPackFn*> run_pack;
79 SidePair<EMat> src;
81 SidePair<PEMat> packed_matrix;
82 SidePair<bool> is_prepacked;
Dtrmul.cc60 SidePair<std::atomic<PackingStatus>*> packing_status, in TrMulTask()
100 SidePair<int> block; in Run()
103 SidePair<int> start, end; in Run()
199 void EnsurePacked(const SidePair<int>& block, const SidePair<int>& start, in EnsurePacked()
200 const SidePair<int>& end, Tuning tuning) { in EnsurePacked()
202 SidePair<int> next_runahead_block{block[Side::kLhs] + 1, in EnsurePacked()
239 SidePair<std::atomic<PackingStatus>*> packing_status_;
244 SidePair<bool*> local_already_packed_;
321 const SidePair<int> origin{0, 0}; in TrMul()
322 const SidePair<int> rounded_dims{packed_lhs.layout.cols, in TrMul()
[all …]
Dside_pair.h40 class SidePair final {
42 SidePair() {} in SidePair() function
43 SidePair(const T& a, const T& b) : elem_{a, b} {} in SidePair() function
Dblock_map.cc38 SidePair<int>* local_pos) { in DecodeTraversalLinear()
44 SidePair<int>* local_pos) { in DecodeTraversalFractalZ()
59 SidePair<int>* local_pos) { in DecodeTraversalFractalU()
70 SidePair<int>* local_pos) { in DecodeTraversalFractalHilbert()
93 SidePair<int>* block) { in GetBlockByIndex()
103 SidePair<int> local_pos; in GetBlockByIndex()
489 void GetBlockMatrixCoords(const BlockMap& block_map, const SidePair<int>& block, in GetBlockMatrixCoords()
490 SidePair<int>* start, SidePair<int>* end) { in GetBlockMatrixCoords()
Dblock_map_test.cc144 int L1Distance(const SidePair<int>& a, const SidePair<int>& b) { in L1Distance()
177 SidePair<int> previous_block_coords(0, 0); in GetBlockByIndexSquareTest()
183 SidePair<int> block_coords; in GetBlockByIndexSquareTest()
Dkernel.h59 static void Run(Tuning tuning, const SidePair<PEMat>& src,
60 const void* mul_params, const SidePair<int>& start,
61 const SidePair<int>& end, EMat* dst) {
82 const SidePair<int>& start, const SidePair<int>& end,