[mlir] Also update inferReturnTensorTypes (NFC)
authorJacques Pienaar <jpienaar@google.com>
Thu, 22 Jul 2021 20:57:36 +0000 (13:57 -0700)
committerJacques Pienaar <jpienaar@google.com>
Thu, 22 Jul 2021 20:57:36 +0000 (13:57 -0700)
commita5b889db7fd3be9e33e29aa4f1a2edfff9a69350
treeb0b6c9565ba5facd77d58fa5404d43bb54848059
parent254c4d174ea3bd3601818b003fc169cdedf24fb9
[mlir] Also update inferReturnTensorTypes (NFC)

Missed this one in the first go.
mlir/include/mlir/Interfaces/InferTypeOpInterface.h
mlir/lib/Interfaces/InferTypeOpInterface.cpp