maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

NFC: Update comments about rank constraints 

These refer to ranked tensors, but are actually applicable to any shaped type

PiperOrigin-RevId: 268931789
This commit is contained in:
Geoffrey Martin-Noble 2019-09-13 10:28:53 -07:00 committed by A. Unique TensorFlower
parent 6cce275301
commit 70653ed600
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/include/mlir/IR/OpBase.td
View File

@ -396,10 +396,10 @@ def F16Tensor : TensorOf<[F16]>;
def F32Tensor : TensorOf<[F32]>;
def F64Tensor : TensorOf<[F64]>;
// Whether a type is a ranked tensor type.
// Whether a shaped type is ranked.
def HasRankPred : CPred<"$_self.cast<ShapedType>().hasRank()">;
// Whether a type is a ranked tensor type with one of the specified ranks.
// Whether a shaped type has one of the specified ranks.
class HasAnyRankOfPred<list<int> ranks> : And<[
HasRankPred,
Or<!foreach(rank, ranks,