Home
last modified time | relevance | path

Searched refs:block_coords (Results 1 – 2 of 2) sorted by relevance

/external/ruy/ruy/
Dblock_map_test.cc183 SidePair<int> block_coords; in GetBlockByIndexSquareTest() local
184 GetBlockByIndex(block_map, block_index, &block_coords); in GetBlockByIndexSquareTest()
185 ++block_hit_counts[block_coords[Side::kLhs] + in GetBlockByIndexSquareTest()
186 num_blocks_per_side * block_coords[Side::kRhs]]; in GetBlockByIndexSquareTest()
187 int distance = L1Distance(block_coords, previous_block_coords); in GetBlockByIndexSquareTest()
190 previous_block_coords = block_coords; in GetBlockByIndexSquareTest()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc4712 const IrArray::Index block_coords = [&] { in EmitTilingKernel() local
4735 b_.CreateICmpEQ(block_coords[i], constant(dims_in_blocks[i] - 1)); in EmitTilingKernel()
4744 std::vector<llvm::Value*> elem_multi_index = block_coords.multidim(); in EmitTilingKernel()
4745 llvm::Type* index_ty = block_coords.GetType(); in EmitTilingKernel()
4748 block_coords[i], in EmitTilingKernel()