Lines Matching refs:init_result
62 init_result = linalg.InitTensorOp([4, 8], f32)
63 return matmul_mono(lhs, rhs, outs=[init_result.result])
76 def test_i8i8i32_matmul(lhs, rhs, init_result): argument
77 return matmul_poly(lhs, rhs, outs=[init_result])
85 def test_i8i8i32_matmul_unsigned(lhs, rhs, init_result): argument
87 lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
100 def test_i8i16i32_matmul(lhs, rhs, init_result): argument
101 return matmul_poly(lhs, rhs, outs=[init_result])
114 def test_i32i32i16_matmul(lhs, rhs, init_result): argument
115 return matmul_poly(lhs, rhs, outs=[init_result])
128 def test_i8i8f32_matmul(lhs, rhs, init_result): argument
129 return matmul_poly(lhs, rhs, outs=[init_result])
137 def test_i8i8f32_matmul_unsigned(lhs, rhs, init_result): argument
139 lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
152 def test_f16f16f32_matmul(lhs, rhs, init_result): argument
153 return matmul_poly(lhs, rhs, outs=[init_result])
166 def test_f64f64f32_matmul(lhs, rhs, init_result): argument
167 return matmul_poly(lhs, rhs, outs=[init_result])