Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dtiled_dot_emitter.cc53 std::vector<llvm::Value*> LoadTile(llvm::Value* minor_dim_offset) const { in LoadTile()
57 result.push_back(vsl_->LoadVector(pointer, minor_dim_offset)); in LoadTile()
67 llvm::Value* minor_dim_offset) const { in StoreTile()
70 vsl_->StoreVector(tile[i], pointers_[i], minor_dim_offset); in StoreTile()
82 llvm::Value* minor_dim_offset, int64 tile_size_along_middle_dim) const { in LoadBroadcastTile() argument
88 pointers_[i], b_->CreateAdd(minor_dim_offset, b_->getInt64(j)))); in LoadBroadcastTile()