Searched defs:init_result (Results 1 – 5 of 5) sorted by relevance
/llvm-project/mlir/test/python/dialects/linalg/opdsl/ |
H A D | emit_misc.py | 62 def test_f32_const(init_result): argument 73 def test_i32_index(init_result): argument 84 def test_f32_elemwise_exp(input, init_result): argument 95 def test_f32_elemwise_log(input, init_result): argument 106 def test_f32_elemwise_abs(input, init_result): argument 117 def test_f32_elemwise_ceil(input, init_result): argument 128 def test_f32_elemwise_floor(input, init_result): argument 139 def test_f32_elemwise_neg(input, init_result): argument 150 def test_c32_elemwise_neg(input, init_result): argument 158 def test_non_default_op_name(input, init_result): argument
|
H A D | emit_matmul.py | 82 def test_i8i8i32_matmul(lhs, rhs, init_result): argument 93 def test_i8i8i32_matmul_unsigned(lhs, rhs, init_result): argument 109 def test_i8i16i32_matmul(lhs, rhs, init_result): argument 125 def test_i32i32i16_matmul(lhs, rhs, init_result): argument 141 test_i8i8f32_matmul(lhs, rhs, init_result) global() argument 152 test_i8i8f32_matmul_unsigned(lhs, rhs, init_result) global() argument 168 test_f16f16f32_matmul(lhs, rhs, init_result) global() argument 184 test_f64f64f32_matmul(lhs, rhs, init_result) global() argument [all...] |
H A D | emit_pooling.py | 55 def test_f32i32_max_pooling(input, shape, init_result): argument 68 def test_f32i32_max_unsigned_pooling(input, shape, init_result): argument 92 def test_f32f32_max_pooling(input, shape, init_result): argument 105 def test_f32i32_min_pooling(input, shape, init_result): argument 123 def test_f32i32_min_unsigned_pooling(input, shape, init_result): argument 141 def test_f32f32_min_pooling(input, shape, init_result): argument
|
H A D | emit_fill.py | 41 def test_fill_0d(value, init_result): argument 49 def test_fill_2d(value, init_result): argument 59 def test_fill_rank_zero_3d(input, init_result): argument
|
H A D | emit_convolution.py | 55 def test_f32i32_conv(input, filter, init_result): argument
|