Searched refs:dim_ids (Results 1 – 2 of 2) sorted by relevance
243 std::vector<int64> dim_ids(num_dims); in QrBlock() local244 std::iota(dim_ids.begin(), dim_ids.end(), 0); in QrBlock()246 /*broadcast_dimensions=*/dim_ids); in QrBlock()
5899 llvm::SmallVector<int64_t, 4> dim_ids(num_dims); in QRBlock() local5900 std::iota(dim_ids.begin(), dim_ids.end(), 0); in QRBlock()5902 loc, a_type, new_x, GetI64ElementsAttr(dim_ids, builder)); in QRBlock()