Unable to lower op 'tensor.from_elements' with multiple parameters

[NeverRaR]

What happened?

I tried to use iree to compile my own ONNX model, but the compilation process aborted in the stream phase because the input ir contains code like:

%from_elements = tensor.from_elements %0, %c2_i32 : tensor<2xi32>

Actually, I noticed that in "ConvertTensorFromElementsPattern", only 'tensor.from_elements' with one parameter are supported for conversion, so:

1. Is this due to some special consideration?
2. How can I handle situations that contain multiple parameters?

Steps to reproduce your issue

There is the smallest test case:

module {
  util.func public @concat_dimension_size(%arg0: tensor<?x?xf32>) -> tensor<2xi32> {
    %c2_i32 = arith.constant 2 : i32
    %c1 = arith.constant 1 : index
    %dim = tensor.dim %arg0, %c1 : tensor<?x?xf32>
    %0 = arith.index_cast %dim : index to i32
    %from_elements = tensor.from_elements %0, %c2_i32 : tensor<2xi32>
    util.return %from_elements : tensor<2xi32>
  }
}

run

iree-compile --mlir-print-ir-after-failure --compile-from=input --compile-to=stream ./test.mlir

then i see:

./test_mlir/from_elements/test1.mlir:7:22: error: failed to legalize operation 'tensor.from_elements' that was explicitly marked illegal
    %from_elements = tensor.from_elements %0, %c2_i32 : tensor<2xi32>
                     ^
./test_mlir/from_elements/test1.mlir:2:3: note: called from
  util.func public @concat_dimension_size(%arg0: tensor<?x?xf32>) -> tensor<2xi32> {
  ^
./test_mlir/from_elements/test1.mlir:7:22: note: see current operation: %3 = "tensor.from_elements"(%2, %0) : (i32, i32) -> tensor<2xi32>
    %from_elements = tensor.from_elements %0, %c2_i32 : tensor<2xi32>
                     ^
// -----// IR Dump After ConvertToStreamPass Failed (iree-stream-conversion) //----- //
Please report issues to https://github.com/openxla/iree/issues and include the crash backtrace.
Stack dump:
0.      Program arguments: ../iree-build/tools/iree-compile --mlir-print-ir-after-failure --compile-from=input --compile-to=stream ./test_mlir/from_elements/test1.mlir

What component(s) does this issue relate to?

Compiler

Version information

iree candidate candidate-20240410.859

Additional context

No response