Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc135 block_sizes_.reserve(req_block_sizes.size());
137 std::back_inserter(block_sizes_),
141 dims_in_blocks_ = ElementWiseCeilOfRatio<int64>(dims_in_tiles_, block_sizes_);
174 DCHECK_EQ(block_index.size(), block_sizes_.size()); in GetTileIndexForBlockOrigin()
176 multidim.reserve(block_sizes_.size()); in GetTileIndexForBlockOrigin()
177 for (int i = 0; i < block_sizes_.size(); ++i) { in GetTileIndexForBlockOrigin()
180 llvm::ConstantInt::get(block_index[i]->getType(), block_sizes_[i]), in GetTileIndexForBlockOrigin()
Dkernel_tiling.h118 return absl::c_accumulate(block_sizes_, 1, std::multiplies<int64>()); in GetNumberOfTilesInOneBlock()
122 return block_sizes_[d]; in GetNumberOfTilesInOneBlockForDimension()
139 absl::Span<const int64> GetBlockSizes() const { return block_sizes_; } in GetBlockSizes()
193 std::vector<int64> block_sizes_; variable
/external/deqp-deps/SPIRV-Tools/source/opt/
Dloop_utils.h37 std::unordered_map<uint32_t, size_t> block_sizes_; member
Dloop_utils.cpp677 block_sizes_.clear(); in Analyze()
688 block_sizes_[bb->id()] = bb_size; in Analyze()
/external/swiftshader/third_party/SPIRV-Tools/source/opt/
Dloop_utils.h37 std::unordered_map<uint32_t, size_t> block_sizes_; member
Dloop_utils.cpp677 block_sizes_.clear(); in Analyze()
688 block_sizes_[bb->id()] = bb_size; in Analyze()