Tai78641 wrote: We are implementing tosa.dim operator that returns a tosa.shape type (which is a new data type internal to tosa dialect) that is backed by I64 values. Will upstream once this is tested out with dynamic shape inference use cases. https://github.com/llvm/llvm-project/pull/77706