Searched refs:multidim_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | ir_array.cc | 80 CHECK_EQ(multidim_.size(), shape.rank()); in Delinearize()
103 : multidim_(shape.rank()), in Index()
112 Delinearize(&multidim_, linear, shape, b); in Index()
118 : multidim_(shape.rank()), in Index()
127 Delinearize(&multidim_, linear, shape, dynamic_dims, b); in Index()
138 : multidim_(multidim.begin(), multidim.end()), in Index()
185 CHECK_EQ(multidim_.size(), output_shape.rank()); in SourceIndexOfReshape()
213 source_multidim_index[i] = multidim_[k]; in SourceIndexOfReshape()
233 Index(absl::Span<llvm::Value* const>(multidim_).subspan( in SourceIndexOfReshape()
270 std::vector<llvm::Value*> source_multi_index(multidim_.size()); in SourceIndexOfSlice()
[all …]
|
| D | ir_array.h | 93 with_offset.multidim_[dim] = in AddOffsetToDim()
94 b->CreateAdd(with_offset.multidim_[dim], addend); in AddOffsetToDim()
98 const std::vector<llvm::Value*>& multidim() const { return multidim_; } in multidim()
190 std::vector<llvm::Value*> multidim_; variable
|