Home
last modified time | relevance | path

Searched refs:pointers_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc40 pointers_.reserve(tile_size_along_major_dim); in MemoryTile()
45 pointers_.push_back(vsl_->ComputeOffsetPointer(matrix, total_offset)); in MemoryTile()
55 result.reserve(pointers_.size()); in LoadTile()
56 for (const auto& pointer : pointers_) { in LoadTile()
68 CHECK_EQ(tile.size(), pointers_.size()); in StoreTile()
69 for (int64 i = 0; i < pointers_.size(); i++) { in StoreTile()
70 vsl_->StoreVector(tile[i], pointers_[i], minor_dim_offset); in StoreTile()
84 result.resize(pointers_.size()); in LoadBroadcastTile()
85 for (int64 i = 0; i < pointers_.size(); i++) { in LoadBroadcastTile()
88 pointers_[i], b_->CreateAdd(minor_dim_offset, b_->getInt64(j)))); in LoadBroadcastTile()
[all …]