Searched refs:multidim_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 72 : multidim_(shape.rank()), in Index()
81 Delinearize(&multidim_, linear, shape, b); in Index()
86 : multidim_(multidim.begin(), multidim.end()), in Index()
144 Index(absl::Span<llvm::Value* const>(multidim_).subspan( in SourceIndexOfReshape()
183 std::vector<llvm::Value*> source_multi_index(multidim_.size()); in SourceIndexOfSlice()
184 for (int i = 0; i < multidim_.size(); ++i) { in SourceIndexOfSlice()
186 auto type = multidim_[i]->getType(); in SourceIndexOfSlice()
190 builder->CreateMul(multidim_[i], in SourceIndexOfSlice()
195 multidim_[i], llvm::ConstantInt::get(type, starts[i])); in SourceIndexOfSlice()
236 builder->CreateMul(multidim_[dimension], in SourceIndexOfBitcast()
[all …]
|
| D | ir_array.h | 67 : multidim_(multidim.begin(), multidim.end()) {
109 const std::vector<llvm::Value*>& multidim() const { return multidim_; } in multidim()
183 std::vector<llvm::Value*> multidim_; variable
|