[Mlir-commits] [mlir] a5b889d - [mlir] Also update inferReturnTensorTypes (NFC)

[Jacques Pienaar]

Author: Jacques Pienaar
Date: 2021-07-22T13:57:36-07:00
New Revision: a5b889db7fd3be9e33e29aa4f1a2edfff9a69350

URL: https://github.com/llvm/llvm-project/commit/a5b889db7fd3be9e33e29aa4f1a2edfff9a69350
DIFF: https://github.com/llvm/llvm-project/commit/a5b889db7fd3be9e33e29aa4f1a2edfff9a69350.diff

LOG: [mlir] Also update inferReturnTensorTypes (NFC)

Missed this one in the first go.

Added: 
    

Modified: 
    mlir/include/mlir/Interfaces/InferTypeOpInterface.h
    mlir/lib/Interfaces/InferTypeOpInterface.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
index 954caa7749bf1..0d6755f211d59 100644
--- a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
+++ b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
@@ -88,7 +88,7 @@ namespace detail {
 // becomes more common.
 LogicalResult inferReturnTensorTypes(
     function_ref<LogicalResult(
-        MLIRContext *, Optional<Location> location, ValueRange operands,
+        MLIRContext *, Optional<Location> location, ValueShapeRange operands,
         DictionaryAttr attributes, RegionRange regions,
         SmallVectorImpl<ShapedTypeComponents> &retComponents)>
         componentTypeFn,

diff  --git a/mlir/lib/Interfaces/InferTypeOpInterface.cpp b/mlir/lib/Interfaces/InferTypeOpInterface.cpp
index cfa2ce486914c..cdae04e563932 100644
--- a/mlir/lib/Interfaces/InferTypeOpInterface.cpp
+++ b/mlir/lib/Interfaces/InferTypeOpInterface.cpp
@@ -22,7 +22,7 @@ namespace mlir {
 
 LogicalResult mlir::detail::inferReturnTensorTypes(
     function_ref<LogicalResult(
-        MLIRContext *, Optional<Location> location, ValueRange operands,
+        MLIRContext *, Optional<Location> location, ValueShapeRange operands,
         DictionaryAttr attributes, RegionRange regions,
         SmallVectorImpl<ShapedTypeComponents> &retComponents)>
         componentTypeFn,