MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetBlockMatrixCoords

Function GetBlockMatrixCoords

tensorflow/lite/experimental/ruy/block_map.cc:208–223  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

206}
207
208void GetBlockMatrixCoords(Side side, const BlockMap& block_map, int block,
209 int* start, int* end) {
210 gemmlowp::ScopedProfilingLabel label("GetBlockMatrixCoords");
211 *start = block * block_map.small_block_dims[side] +
212 std::min(block, block_map.large_blocks[side]) *
213 block_map.kernel_dims[side];
214 *end =
215 *start + block_map.small_block_dims[side] +
216 (block < block_map.large_blocks[side] ? block_map.kernel_dims[side] : 0);
217
218 RUY_DCHECK_EQ(0, *start % block_map.kernel_dims[side]);
219 RUY_DCHECK_EQ(0, *end % block_map.kernel_dims[side]);
220 RUY_DCHECK_LE(*end, block_map.dims[side]);
221 RUY_DCHECK_LT(*start, *end);
222 RUY_DCHECK_GE(*start, 0);
223}
224
225void GetBlockMatrixCoords(const BlockMap& block_map, const SidePair<int>& block,
226 SidePair<int>* start, SidePair<int>* end) {

Callers 3

DumpFunction · 0.85
RunMethod · 0.85
EnsurePackedMethod · 0.85

Calls 1

minFunction · 0.50

Tested by

no test coverage detected