Lines Matching refs:f64

54         ^bb(%a: f64, %b: f64, %x: f64):
55 %1 = sparse_tensor.binary %a, %b : f64, f64 to f64
57 ^bb0(%a0: f64, %b0: f64):
58 %cmp = arith.cmpf "olt", %a0, %b0 : f64
59 %2 = arith.select %cmp, %a0, %b0: f64
60 sparse_tensor.yield %2 : f64
64 linalg.yield %1 : f64
79 ^bb(%a: f64, %b: f64, %x: f64):
80 %1 = sparse_tensor.binary %a, %b : f64, f64 to f64
82 ^bb0(%a0: f64, %b0: f64):
83 %ret = arith.mulf %a0, %b0 : f64
84 sparse_tensor.yield %ret : f64
88 linalg.yield %1 : f64
103 ^bb(%a: f64, %b: f64, %x: f64):
104 %1 = sparse_tensor.binary %a, %b : f64, f64 to f64
108 linalg.yield %1 : f64
121 ^bb(%a: f64, %x: i32):
123 %1 = sparse_tensor.binary %a, %idx : f64, index to i32
125 ^bb0(%x0: f64, %i: index):
148 ^bb(%a: f64, %b: f64, %x: f64):
149 %1 = sparse_tensor.binary %a, %b: f64, f64 to f64
151 ^bb0(%x0: f64, %y0: f64):
152 %ret = arith.addf %x0, %y0 : f64
153 sparse_tensor.yield %ret : f64
157 ^bb0(%x1: f64):
158 %lret = arith.negf %x1 : f64
159 sparse_tensor.yield %lret : f64
161 linalg.yield %1 : f64
174 ^bb0(%a: f64, %b: f64, %c: f64) :
175 %result = sparse_tensor.binary %a, %b : f64, f64 to f64
177 ^bb0(%x: f64, %y: f64):
178 %ret = arith.addf %x, %y : f64
179 sparse_tensor.yield %ret : f64
183 linalg.yield %result : f64
196 ^bb0(%a: f64, %b: f64, %c: f64) :
197 %result = sparse_tensor.binary %a, %b : f64, f64 to f64
199 ^bb0(%x: f64, %y: f64):
200 %ret = arith.addf %x, %y : f64
201 sparse_tensor.yield %ret : f64
204 ^bb0(%x: f64):
205 sparse_tensor.yield %x : f64
208 ^bb0(%y: f64):
209 sparse_tensor.yield %y : f64
211 linalg.yield %result : f64
224 ^bb0(%a: f64, %b: f64, %c: f64) :
227 %result = sparse_tensor.binary %a, %b : f64, f64 to f64
229 ^bb0(%x: f64, %y: f64):
231 %upperTriangleResult = arith.addf %x, %y : f64
232 %lowerTriangleResult = arith.subf %x, %y : f64
233 %ret = arith.select %cmp, %upperTriangleResult, %lowerTriangleResult : f64
234 sparse_tensor.yield %ret : f64
238 ^bb0(%y: f64):
240 %lowerTriangleResult = arith.negf %y : f64
241 %ret = arith.select %cmp, %y, %lowerTriangleResult : f64
242 sparse_tensor.yield %ret : f64
244 linalg.yield %result : f64
254 %thres_out_up = arith.constant 2.0 : f64
255 %thres_out_lo = arith.constant -2.0 : f64
261 ^bb0(%a: f64, %b: f64, %c: f64) :
262 %result = sparse_tensor.binary %a, %b : f64, f64 to f64
264 ^bb0(%x: f64, %y: f64):
266 %thres_up = arith.constant 1.0 : f64
267 %thres_lo = arith.constant -1.0 : f64
268 %result = arith.subf %x, %y : f64
269 %cmp = arith.cmpf "oge", %result, %thres_up : f64
270 %tmp = arith.select %cmp, %thres_up, %result : f64
271 %cmp1 = arith.cmpf "ole", %tmp, %thres_lo : f64
272 %ret = arith.select %cmp1, %thres_lo, %tmp : f64
273 sparse_tensor.yield %ret : f64
276 ^bb0(%x: f64):
278 %cmp = arith.cmpf "oge", %x, %thres_out_up : f64
279 %tmp = arith.select %cmp, %thres_out_up, %x : f64
280 %cmp1 = arith.cmpf "ole", %tmp, %thres_out_lo : f64
281 %ret = arith.select %cmp1, %thres_out_lo, %tmp : f64
282 sparse_tensor.yield %ret : f64
285 ^bb0(%y: f64):
286 %ny = arith.negf %y : f64
287 %cmp = arith.cmpf "oge", %ny, %thres_out_up : f64
288 %tmp = arith.select %cmp, %thres_out_up, %ny : f64
289 %cmp1 = arith.cmpf "ole", %tmp, %thres_out_lo : f64
290 %ret = arith.select %cmp1, %thres_out_lo, %tmp : f64
291 sparse_tensor.yield %ret : f64
293 linalg.yield %result : f64
306 ^bb0(%a: f64, %b: f64, %c: i8) :
307 %result = sparse_tensor.binary %a, %b : f64, f64 to i8
309 ^bb0(%x: f64, %y: f64):
310 %cmp = arith.cmpf "oeq", %x, %y : f64
329 ^bb0(%a: f64, %b: f64, %c: f64) :
330 %result = sparse_tensor.binary %a, %b : f64, f64 to f64
334 ^bb0(%y: f64):
335 %ret = arith.negf %y : f64
336 sparse_tensor.yield %ret : f64
338 linalg.yield %result : f64
350 %d0 = arith.constant -1.0 : f64
376 %d0 = arith.constant 0.0 : f64
386 %du = arith.constant -1.0 : f64