Searched refs:MemoryTile (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 30 class MemoryTile { class
36 MemoryTile(VectorSupportLibrary* vsl, llvm::IRBuilder<>* b, in MemoryTile() function in xla::cpu::__anon016ac5180111::MemoryTile
256 MemoryTile GetLhsMemoryTile(llvm::Value* column_start, int64 column_count) { in GetLhsMemoryTile()
257 return MemoryTile(&vsl_, b_, /*matrix=*/lhs_, in GetLhsMemoryTile()
275 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile,
294 MemoryTile lhs_memory_tile = GetLhsMemoryTile(/*column_start=*/column, in EmitOuterLoopBody()
322 MemoryTile* lhs_memory_tile, const std::vector<llvm::Value*>& rhs_tile, in EmitInnerLoopTiled()
478 MemoryTile GetLhsMemoryTile(llvm::Value* row_start, int64 row_count) { in GetLhsMemoryTile()
479 return MemoryTile(&vsl_, b_, /*matrix=*/lhs_, in GetLhsMemoryTile()
487 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile, int64 rows,
[all …]
|