Searched refs:vsl_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | tiled_dot_emitter.cc | 39 : vsl_(vsl), b_(b) { in MemoryTile()
45 pointers_.push_back(vsl_->ComputeOffsetPointer(matrix, total_offset)); in MemoryTile()
57 result.push_back(vsl_->LoadVector(pointer, minor_dim_offset)); in LoadTile()
70 vsl_->StoreVector(tile[i], pointers_[i], minor_dim_offset); in StoreTile()
87 result[i].push_back(vsl_->LoadBroadcast( in LoadBroadcastTile()
95 VectorSupportLibrary* vsl_; member in xla::cpu::__anon016ac5180111::MemoryTile
243 vsl_(config.scalar_type(), /*vector_size=*/config.tile_rows(), b_, "") { in ColumnMajorMatrixVectorProductEmitter()
257 return MemoryTile(&vsl_, b_, /*matrix=*/lhs_, in GetLhsMemoryTile()
266 llvm::Value* base_pointer = vsl_.ComputeOffsetPointer(rhs_, offset); in LoadRhsTile()
270 result.push_back(vsl_.LoadBroadcast(base_pointer, i)); in LoadRhsTile()
[all …]
|