tree f89f960b12a3b7f4f5ef782bf1e150051e722498
parent d3b0b1963af778d9fc8707eebd2e315edc915286
author Benjamin Kramer <kramerb@google.com> 1624363122 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1624363537 -0700

[TF:mhlo] Use tensor.from_elements directly when lowering tf.ExpandDims

tensor.from_elements saves one instruction and has better support as it doesn't
introduce a !shape.shape type where all other shapes are index tensors.

PiperOrigin-RevId: 380782976
Change-Id: Idf463d4b320fe8af521a088a878a5f5b487a4e74