Lines Matching refs:sparse_tensor
3 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
7 // CHECK: %[[T:.*]] = sparse_tensor.new %[[A]] : !llvm.ptr<i8> to tensor<128xf64, #{{.*}}>
10 %0 = sparse_tensor.new %arg0 : !llvm.ptr<i8> to tensor<128xf64, #SparseVector>
16 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
29 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
33 // CHECK: %[[T:.*]] = sparse_tensor.convert %[[A]] : tensor<64xf32> to tensor<64xf32, #{{.*}}>
36 %0 = sparse_tensor.convert %arg0 : tensor<64xf32> to tensor<64xf32, #SparseVector>
42 #SparseTensor = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "dense", "compressed" ] }>
46 // CHECK: %[[T:.*]] = sparse_tensor.convert %[[A]] : tensor<8x8x8xf64, #{{.*}}> to tensor<8x8…
49 %0 = sparse_tensor.convert %arg0 : tensor<8x8x8xf64, #SparseTensor> to tensor<8x8x8xf64>
55 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
60 // CHECK: %[[T:.*]] = sparse_tensor.pointers %[[A]], %[[C]] : tensor<128xf64, #{{.*}}> to mem…
64 %0 = sparse_tensor.pointers %arg0, %c : tensor<128xf64, #SparseVector> to memref<?xindex>
70 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
75 // CHECK: %[[T:.*]] = sparse_tensor.indices %[[A]], %[[C]] : tensor<128xf64, #{{.*}}> to memr…
79 %0 = sparse_tensor.indices %arg0, %c : tensor<128xf64, #SparseVector> to memref<?xindex>
85 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
89 // CHECK: %[[T:.*]] = sparse_tensor.values %[[A]] : tensor<128xf64, #{{.*}}> to memref<?xf64>
92 %0 = sparse_tensor.values %arg0 : tensor<128xf64, #SparseVector> to memref<?xf64>
98 #DenseMatrix = #sparse_tensor.encoding<{dimLevelType = ["dense","dense"]}>
102 // CHECK: %[[T:.*]] = sparse_tensor.load %[[A]] : tensor<16x32xf64, #{{.*}}>
105 %0 = sparse_tensor.load %arg0 : tensor<16x32xf64, #DenseMatrix>
111 #DenseMatrix = #sparse_tensor.encoding<{dimLevelType = ["dense","dense"]}>
115 // CHECK: %[[T:.*]] = sparse_tensor.load %[[A]] hasInserts : tensor<16x32xf64, #{{.*}}>
118 %0 = sparse_tensor.load %arg0 hasInserts : tensor<16x32xf64, #DenseMatrix>
124 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
127 // CHECK-SAME: %[[A:.*]]: tensor<128xf64, #sparse_tensor.encoding<{{.*}}>>,
130 // CHECK: sparse_tensor.lex_insert %[[A]], %[[B]], %[[C]] : tensor<128xf64, #{{.*}}>, memref<…
133 …sparse_tensor.lex_insert %arg0, %arg1, %arg2 : tensor<128xf64, #SparseVector>, memref<?xindex>, f64
139 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
142 // CHECK-SAME: %[[A:.*]]: tensor<8x8xf64, #sparse_tensor.encoding<{{.*}}>>)
143 // CHECK: sparse_tensor.expand %[[A]]
146 %values, %filled, %added, %count = sparse_tensor.expand %arg0
153 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
156 // CHECK-SAME: %[[A:.*]]: tensor<8x8xf64, #sparse_tensor.encoding<{{.*}}>>,
157 // CHECK: sparse_tensor.compress %[[A]]
162 sparse_tensor.compress %arg0, %arg1, %arg2, %arg3, %arg4, %arg5
169 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
172 // CHECK-SAME: %[[A:.*]]: tensor<?x?xf64, #sparse_tensor.encoding<{{.*}}>>,
174 // CHECK: sparse_tensor.out %[[A]], %[[B]] : tensor<?x?xf64, #sparse_tensor.encoding<{{.*}}>>…
177 sparse_tensor.out %arg0, %arg1 : tensor<?x?xf64, #SparseMatrix>, !llvm.ptr<i8>
183 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
188 // CHECK: %[[C1:.*]] = sparse_tensor.binary %[[A]], %[[B]] : f64, i64 to f64
191 // CHECK: sparse_tensor.yield %[[A1]] : f64
196 // CHECK: sparse_tensor.yield %[[Z]] : f64
202 %r = sparse_tensor.binary %arg0, %arg1 : f64, i64 to f64
205 sparse_tensor.yield %x : f64
210 sparse_tensor.yield %cf0 : f64
217 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
221 // CHECK: %[[C1:.*]] = sparse_tensor.unary %[[A]] : f64 to f64
224 // CHECK: sparse_tensor.yield %[[A1]] : f64
228 // CHECK: sparse_tensor.yield %[[R]] : f64
233 %r = sparse_tensor.unary %arg0 : f64 to f64
236 sparse_tensor.yield %x : f64
240 sparse_tensor.yield %cf1 : f64
247 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
251 // CHECK: %[[C1:.*]] = sparse_tensor.unary %[[A]] : f64 to i64
255 // CHECK: sparse_tensor.yield %[[R]] : i64
262 %r = sparse_tensor.unary %arg0 : f64 to i64
266 sparse_tensor.yield %ret : i64
274 #SparseMatrix = #sparse_tensor.encoding<{dimLevelType = ["compressed", "compressed"]}>
279 // CHECK: %[[C1:.*]] = sparse_tensor.reduce %[[A]], %[[B]], %[[Z]] : f64 {
281 // CHECK: sparse_tensor.yield %[[A1]] : f64
287 %r = sparse_tensor.reduce %arg0, %arg1, %cf0 : f64 {
289 sparse_tensor.yield %x : f64