| /llvm-project-15.0.7/mlir/test/python/dialects/linalg/opdsl/ |
| H A D | emit_misc.py | 59 def test_f32_const(init_result): argument
60 return test_const(outs=[init_result])
70 def test_i32_index(init_result): argument
71 return test_index(outs=[init_result])
80 def test_f32_elemwise_exp(input, init_result): argument
90 def test_f32_elemwise_log(input, init_result): argument
100 def test_f32_elemwise_abs(input, init_result): argument
110 def test_f32_elemwise_ceil(input, init_result): argument
120 def test_f32_elemwise_floor(input, init_result): argument
130 def test_f32_elemwise_neg(input, init_result): argument
[all …]
|
| H A D | emit_matmul.py | 62 init_result = linalg.InitTensorOp([4, 8], f32)
76 def test_i8i8i32_matmul(lhs, rhs, init_result): argument
77 return matmul_poly(lhs, rhs, outs=[init_result])
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])
152 def test_f16f16f32_matmul(lhs, rhs, init_result): argument
[all …]
|
| H A D | emit_pooling.py | 53 def test_f32i32_max_pooling(input, shape, init_result): argument
55 input, shape, outs=[init_result], strides=[2, 4], dilations=[1, 2])
64 def test_f32i32_max_unsigned_pooling(input, shape, init_result): argument
68 outs=[init_result],
86 def test_f32f32_max_pooling(input, shape, init_result): argument
97 def test_f32i32_min_pooling(input, shape, init_result): argument
101 outs=[init_result],
113 def test_f32i32_min_unsigned_pooling(input, shape, init_result): argument
117 outs=[init_result],
129 def test_f32f32_min_pooling(input, shape, init_result): argument
[all …]
|
| H A D | emit_fill.py | 39 def test_fill_0d(value, init_result): argument
40 return fill_poly(value, outs=[init_result])
47 def test_fill_2d(value, init_result): argument
48 return fill_poly(value, outs=[init_result])
56 def test_fill_rank_zero_3d(input, init_result): argument
57 return fill_rank_zero_poly(input, outs=[init_result])
|
| H A D | emit_convolution.py | 53 def test_f32i32_conv(input, filter, init_result): argument
56 input, filter, outs=[init_result], strides=[2, 4])
|
| /llvm-project-15.0.7/mlir/test/python/dialects/linalg/ |
| H A D | ops.py | 99 init_result = linalg.InitTensorOp([4, 8], f32)
105 unary_result = linalg.elemwise_unary(lhs, outs=[init_result.result])
114 outs=[init_result.result],
134 init_result = linalg.InitTensorOp([4, 8], f32)
143 return linalg.matmul(lhs, rhs, outs=[init_result.result])
160 init_result = linalg.InitTensorOp([4, 8], f32)
163 lhs, rhs, outs=[init_result.result], emit_generic=True)
|