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::__anon8d179a540111::MemoryTile
258 MemoryTile GetLhsMemoryTile(llvm::Value* column_start, int64_t column_count) { in GetLhsMemoryTile()
259 return MemoryTile(&vsl_, b_, /*matrix=*/lhs_, in GetLhsMemoryTile()
277 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile,
296 MemoryTile lhs_memory_tile = GetLhsMemoryTile(/*column_start=*/column, in EmitOuterLoopBody()
324 MemoryTile* lhs_memory_tile, const std::vector<llvm::Value*>& rhs_tile, in EmitInnerLoopTiled()
482 MemoryTile GetLhsMemoryTile(llvm::Value* row_start, int64_t row_count) { in GetLhsMemoryTile()
483 return MemoryTile(&vsl_, b_, /*matrix=*/lhs_, in GetLhsMemoryTile()
491 void EmitInnerLoopTiled(MemoryTile* lhs_memory_tile, int64_t rows,
[all …]
|