Lines Matching refs:f64

117 func.func @sparse_unannotated_insert(%arg0: tensor<128xf64>, %arg1: memref<?xindex>, %arg2: f64) {
119   sparse_tensor.lex_insert %arg0, %arg1, %arg2 : tensor<128xf64>, memref<?xindex>, f64
180 func.func @invalid_binary_num_args_mismatch_overlap(%arg0: f64, %arg1: f64) -> f64 {
182   %r = sparse_tensor.binary %arg0, %arg1 : f64, f64 to f64
184       ^bb0(%x: f64):
185         sparse_tensor.yield %x : f64
189   return %r : f64
194 func.func @invalid_binary_num_args_mismatch_right(%arg0: f64, %arg1: f64) -> f64 {
196   %r = sparse_tensor.binary %arg0, %arg1 : f64, f64 to f64
200       ^bb0(%x: f64, %y: f64):
201         sparse_tensor.yield %y : f64
203   return %r : f64
208 func.func @invalid_binary_argtype_mismatch(%arg0: f64, %arg1: f64) -> f64 {
210   %r = sparse_tensor.binary %arg0, %arg1 : f64, f64 to f64
212       ^bb0(%x: f64, %y: f32):
213         sparse_tensor.yield %x : f64
217   return %r : f64
222 func.func @invalid_binary_wrong_return_type(%arg0: f64, %arg1: f64) -> f64 {
224   %0 = sparse_tensor.binary %arg0, %arg1 : f64, f64 to f64
227       ^bb0(%x: f64):
232   return %0 : f64
237 func.func @invalid_binary_wrong_identity_type(%arg0: i64, %arg1: f64) -> f64 {
239   %0 = sparse_tensor.binary %arg0, %arg1 : i64, f64 to f64
243   return %0 : f64
248 func.func @invalid_binary_wrong_yield(%arg0: f64, %arg1: f64) -> f64 {
250   %0 = sparse_tensor.binary %arg0, %arg1 : f64, f64 to f64
253       ^bb0(%x: f64):
254         tensor.yield %x : f64
257   return %0 : f64
262 func.func @invalid_unary_argtype_mismatch(%arg0: f64) -> f64 {
264   %r = sparse_tensor.unary %arg0 : f64 to f64
270   return %r : f64
275 func.func @invalid_unary_num_args_mismatch(%arg0: f64) -> f64 {
277   %r = sparse_tensor.unary %arg0 : f64 to f64
280       ^bb0(%x: f64):
281         sparse_tensor.yield %x : f64
283   return %r : f64
288 func.func @invalid_unary_wrong_return_type(%arg0: f64) -> f64 {
290   %0 = sparse_tensor.unary %arg0 : f64 to f64
292       ^bb0(%x: f64):
297   return %0 : f64
302 func.func @invalid_unary_wrong_yield(%arg0: f64) -> f64 {
304   %0 = sparse_tensor.unary %arg0 : f64 to f64
306       ^bb0(%x: f64):
307         tensor.yield %x : f64
310   return %0 : f64
315 func.func @invalid_reduce_num_args_mismatch(%arg0: f64, %arg1: f64) -> f64 {
316   %cf1 = arith.constant 1.0 : f64
318   %r = sparse_tensor.reduce %arg0, %arg1, %cf1 : f64 {
319       ^bb0(%x: f64):
320         sparse_tensor.yield %x : f64
322   return %r : f64
331       ^bb0(%x: f64, %y: f64):
340 func.func @invalid_reduce_return_type_mismatch(%arg0: f64, %arg1: f64) -> f64 {
341   %cf1 = arith.constant 1.0 : f64
343   %r = sparse_tensor.reduce %arg0, %arg1, %cf1 : f64 {
344       ^bb0(%x: f64, %y: f64):
348   return %r : f64
353 func.func @invalid_reduce_wrong_yield(%arg0: f64, %arg1: f64) -> f64 {
354   %cf1 = arith.constant 1.0 : f64
356   %r = sparse_tensor.reduce %arg0, %arg1, %cf1 : f64 {
357       ^bb0(%x: f64, %y: f64):
361   return %r : f64