HomeSort by relevance Sort by last modified time
    Searched refs:block_sizes_ (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
kernel_tiling.h 118 return absl::c_accumulate(block_sizes_, 1, std::multiplies<int64>());
122 return block_sizes_[d];
139 absl::Span<const int64> GetBlockSizes() const { return block_sizes_; }
193 std::vector<int64> block_sizes_; member in class:xla::llvm_ir::KernelMappingScheme
195 // dims_in_tile_ and block_sizes_.
kernel_tiling.cc 135 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());
176 multidim.reserve(block_sizes_.size());
177 for (int i = 0; i < block_sizes_.size(); ++i) {
180 llvm::ConstantInt::get(block_index[i]->getType(), block_sizes_[i]),
  /external/deqp-deps/SPIRV-Tools/source/opt/
loop_utils.h 37 std::unordered_map<uint32_t, size_t> block_sizes_; member in struct:spvtools::opt::CodeMetrics
loop_utils.cpp 677 block_sizes_.clear();
688 block_sizes_[bb->id()] = bb_size;
  /external/swiftshader/third_party/SPIRV-Tools/source/opt/
loop_utils.h 37 std::unordered_map<uint32_t, size_t> block_sizes_; member in struct:spvtools::opt::CodeMetrics
loop_utils.cpp 677 block_sizes_.clear();
688 block_sizes_[bb->id()] = bb_size;

Completed in 2535 milliseconds