Searched refs:lhs_memory_tile (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 275 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile,
294 MemoryTile lhs_memory_tile = GetLhsMemoryTile(/*column_start=*/column, in EmitOuterLoopBody() local
299 EmitInnerLoopTiled(&lhs_memory_tile, rhs_tile, in EmitOuterLoopBody()
322 MemoryTile* lhs_memory_tile, const std::vector<llvm::Value*>& rhs_tile, in EmitInnerLoopTiled() argument
329 lhs_memory_tile->LoadTile(/*minor_dim_offset=*/row); in EmitInnerLoopTiled()
487 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile, int64 rows,
505 MemoryTile lhs_memory_tile = GetLhsMemoryTile(/*row_start=*/row, in EmitOuterLoopBody() local
513 EmitInnerLoopTiled(&lhs_memory_tile, /*rows=*/row_count, in EmitOuterLoopBody()
563 MemoryTile* lhs_memory_tile, int64 rows, in EmitInnerLoopTiled() argument
570 lhs_memory_tile->LoadTile(/*minor_dim_offset=*/col); in EmitInnerLoopTiled()
[all …]
|