Is milo::BroadcastInDimOp supported non constant parameter or dynamic shape?

[mofheka]

Such as using a Value type for dims parameter, and a dynamic shape for result type parameter of BroadcastInDimOp.

[Geoffrey Martin-Noble]

You can see from the definition of the op that it requires these to be static. There is a dynamic variant that accepts a Value for its output dimensions. It still requires a static mapping between the dimensions of the input and output, however (which dimension in the input corresponds to each dimension in the output).

On Wed, Mar 31, 2021 at 1:24 AM mofheka <mofh...@gmail.com> wrote:

Such as using a Value type for dims parameter, and a dynamic shape for result type parameter of BroadcastInDimOp.

--
You received this message because you are subscribed to the Google Groups "MLIR" group.
To unsubscribe from this group and stop receiving emails from it, send an email to mlir+uns...@tensorflow.org.
To view this discussion on the web visit https://groups.google.com/a/tensorflow.org/d/msgid/mlir/61aec6fa-5cba-4c3e-90ff-36dbdd54cc3en%40tensorflow.org.