projects / platform / upstream / llvm.git / commitdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | patch | inline | side by side (parent: 254c4d1)
[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)
Missed this one in the first go.

mlir/include/mlir/Interfaces/InferTypeOpInterface.h patch | blob | history
mlir/lib/Interfaces/InferTypeOpInterface.cpp patch | blob | history

diff --git a/mlir/include/mlir/Interfaces/InferTypeOpInterface.h b/mlir/include/mlir/Interfaces/InferTypeOpInterface.h
index 954caa7..0d6755f 100644 (file)
--- 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 cfa2ce4..cdae04e 100644 (file)
--- 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,
Domain: System / Toolchain;