Lines Matching refs:f64
129 // CHECK-SAME: %[[C:.*]]: f64) {
130 …K: sparse_tensor.lex_insert %[[A]], %[[B]], %[[C]] : tensor<128xf64, #{{.*}}>, memref<?xindex>, f64
132 func.func @sparse_insert(%arg0: tensor<128xf64, #SparseVector>, %arg1: memref<?xindex>, %arg2: f64)…
133 …sparse_tensor.lex_insert %arg0, %arg1, %arg2 : tensor<128xf64, #SparseVector>, memref<?xindex>, f64
186 // CHECK-SAME: %[[A:.*]]: f64, %[[B:.*]]: i64) -> f64 {
187 // CHECK: %[[Z:.*]] = arith.constant 0.000000e+00 : f64
188 // CHECK: %[[C1:.*]] = sparse_tensor.binary %[[A]], %[[B]] : f64, i64 to f64
190 // CHECK: ^bb0(%[[A1:.*]]: f64, %[[B1:.*]]: i64):
191 // CHECK: sparse_tensor.yield %[[A1]] : f64
196 // CHECK: sparse_tensor.yield %[[Z]] : f64
198 // CHECK: return %[[C1]] : f64
200 func.func @sparse_binary(%arg0: f64, %arg1: i64) -> f64 {
201 %cf0 = arith.constant 0.0 : f64
202 %r = sparse_tensor.binary %arg0, %arg1 : f64, i64 to f64
204 ^bb0(%x: f64, %y: i64):
205 sparse_tensor.yield %x : f64
210 sparse_tensor.yield %cf0 : f64
212 return %r : f64
220 // CHECK-SAME: %[[A:.*]]: f64) -> f64 {
221 // CHECK: %[[C1:.*]] = sparse_tensor.unary %[[A]] : f64 to f64
223 // CHECK: ^bb0(%[[A1:.*]]: f64):
224 // CHECK: sparse_tensor.yield %[[A1]] : f64
227 // CHECK: %[[R:.*]] = arith.constant -1.000000e+00 : f64
228 // CHECK: sparse_tensor.yield %[[R]] : f64
230 // CHECK: return %[[C1]] : f64
232 func.func @sparse_unary(%arg0: f64) -> f64 {
233 %r = sparse_tensor.unary %arg0 : f64 to f64
235 ^bb0(%x: f64):
236 sparse_tensor.yield %x : f64
239 %cf1 = arith.constant -1.0 : f64
240 sparse_tensor.yield %cf1 : f64
242 return %r : f64
250 // CHECK-SAME: %[[A:.*]]: f64) -> i64 {
251 // CHECK: %[[C1:.*]] = sparse_tensor.unary %[[A]] : f64 to i64
253 // CHECK: ^bb0(%[[A1:.*]]: f64):
254 // CHECK: %[[R:.*]] = arith.fptosi %[[A1]] : f64 to i64
261 func.func @sparse_unary(%arg0: f64) -> i64 {
262 %r = sparse_tensor.unary %arg0 : f64 to i64
264 ^bb0(%x: f64):
265 %ret = arith.fptosi %x : f64 to i64
277 // CHECK-SAME: %[[A:.*]]: f64, %[[B:.*]]: f64) -> f64 {
278 // CHECK: %[[Z:.*]] = arith.constant 0.000000e+00 : f64
279 // CHECK: %[[C1:.*]] = sparse_tensor.reduce %[[A]], %[[B]], %[[Z]] : f64 {
280 // CHECK: ^bb0(%[[A1:.*]]: f64, %[[B1:.*]]: f64):
281 // CHECK: sparse_tensor.yield %[[A1]] : f64
283 // CHECK: return %[[C1]] : f64
285 func.func @sparse_reduce_2d_to_1d(%arg0: f64, %arg1: f64) -> f64 {
286 %cf0 = arith.constant 0.0 : f64
287 %r = sparse_tensor.reduce %arg0, %arg1, %cf0 : f64 {
288 ^bb0(%x: f64, %y: f64):
289 sparse_tensor.yield %x : f64
291 return %r : f64