| ir_array.h | 63 // Constructs an index from multi-dimensional index "multidim". The linear
65 explicit Index(absl::Span<llvm::Value* const> multidim,
67 : multidim_(multidim.begin(), multidim.end()) {
71 for (const auto* dim : multidim) {
74 index_type_ = multidim[0]->getType();
80 CHECK(absl::c_all_of(multidim, [&](llvm::Value* v) {
98 Index(absl::Span<llvm::Value* const> multidim, const Shape& shape,
104 std::vector<llvm::Value*> multi_index = multidim();
109 const std::vector<llvm::Value*>& multidim() const { return multidim_; function in class:xla::llvm_ir::IrArray::Index
[all...] |
