Lines Matching defs:init_result
66 init_result = tensor.empty([4, 8], f32)
67 return matmul_mono(lhs, rhs, outs=[init_result])
82 def test_i8i8i32_matmul(lhs, rhs, init_result):
83 return matmul_poly(lhs, rhs, outs=[init_result])
93 def test_i8i8i32_matmul_unsigned(lhs, rhs, init_result):
94 return matmul_poly(lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
109 def test_i8i16i32_matmul(lhs, rhs, init_result):
110 return matmul_poly(lhs, rhs, outs=[init_result])
125 def test_i32i32i16_matmul(lhs, rhs, init_result):
126 return matmul_poly(lhs, rhs, outs=[init_result])
141 def test_i8i8f32_matmul(lhs, rhs, init_result):
142 return matmul_poly(lhs, rhs, outs=[init_result])
152 def test_i8i8f32_matmul_unsigned(lhs, rhs, init_result):
153 return matmul_poly(lhs, rhs, outs=[init_result], cast=TypeFn.cast_unsigned)
168 def test_f16f16f32_matmul(lhs, rhs, init_result):
169 return matmul_poly(lhs, rhs, outs=[init_result])
184 def test_f64f64f32_matmul(lhs, rhs, init_result):
185 return matmul_poly(lhs, rhs, outs=[init_result])