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 | 277 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile,
296 MemoryTile lhs_memory_tile = GetLhsMemoryTile(/*column_start=*/column, in EmitOuterLoopBody() local
301 EmitInnerLoopTiled(&lhs_memory_tile, rhs_tile, in EmitOuterLoopBody()
324 MemoryTile* lhs_memory_tile, const std::vector<llvm::Value*>& rhs_tile, in EmitInnerLoopTiled() argument
331 lhs_memory_tile->LoadTile(/*minor_dim_offset=*/row); in EmitInnerLoopTiled()
491 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile, int64_t rows,
509 MemoryTile lhs_memory_tile = GetLhsMemoryTile(/*row_start=*/row, in EmitOuterLoopBody() local
519 EmitInnerLoopTiled(&lhs_memory_tile, /*rows=*/row_count, in EmitOuterLoopBody()
569 MemoryTile* lhs_memory_tile, int64_t rows, in EmitInnerLoopTiled() argument
576 lhs_memory_tile->LoadTile(/*minor_dim_offset=*/col); in EmitInnerLoopTiled()
[all …]
|