Lines Matching refs:arg1

4 func.func @pow_noop(%arg0: f32, %arg1 : vector<4xf32>) -> (f32, vector<4xf32>) {
5   // CHECK: return %arg0, %arg1
9   %1 = math.powf %arg1, %v : vector<4xf32>
14 func.func @pow_square(%arg0: f32, %arg1 : vector<4xf32>) -> (f32, vector<4xf32>) {
16   // CHECK: %[[VECTOR:.*]] = arith.mulf %arg1, %arg1
21   %1 = math.powf %arg1, %v : vector<4xf32>
26 func.func @pow_cube(%arg0: f32, %arg1 : vector<4xf32>) -> (f32, vector<4xf32>) {
29   // CHECK: %[[TMP_V:.*]] = arith.mulf %arg1, %arg1
30   // CHECK: %[[VECTOR:.*]] = arith.mulf %arg1, %[[TMP_V]]
35   %1 = math.powf %arg1, %v : vector<4xf32>
40 func.func @pow_recip(%arg0: f32, %arg1 : vector<4xf32>) -> (f32, vector<4xf32>) {
44   // CHECK: %[[VECTOR:.*]] = arith.divf %[[CST_V]], %arg1
49   %1 = math.powf %arg1, %v : vector<4xf32>
54 func.func @pow_sqrt(%arg0: f32, %arg1 : vector<4xf32>) -> (f32, vector<4xf32>) {
56   // CHECK: %[[VECTOR:.*]] = math.sqrt %arg1
61   %1 = math.powf %arg1, %v : vector<4xf32>
66 func.func @pow_rsqrt(%arg0: f32, %arg1 : vector<4xf32>) -> (f32, vector<4xf32>) {
68   // CHECK: %[[VECTOR:.*]] = math.rsqrt %arg1
73   %1 = math.powf %arg1, %v : vector<4xf32>