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

TensorFlow operands can be broadcasted with each other and with the result, relax the verifier. 

PiperOrigin-RevId: 254206814
This commit is contained in:
Mehdi Amini 2019-06-20 08:58:24 -07:00 committed by jpienaar
parent 2d6bbabe4b
commit 25050960af
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1137,8 +1137,9 @@ class TCOpIsBroadcastableToRes<int opId, int resId> : And<[
IsShapedTypePred>,
SubstLeaves<"$_self", "$_op.getResult(" # resId # ")->getType()",
IsShapedTypePred>,
CPred<"OpTrait::util::getBroadcastedType($_op.getOperand(" # opId # "), "
"$_op.getResult(" # resId # "))">]>;
CPred<"OpTrait::util::getBroadcastedType("
"$_op.getOperand(" # opId # ")->getType(), "
"$_op.getResult(" # resId # ")->getType())">]>;
// Predicate to verify that all the operands at the given `indices`
// have the same element type.