From dd5895d08313c5f122ff456cd6ca54e21d7830d9 Mon Sep 17 00:00:00 2001 From: Stephan Herhut Date: Tue, 8 Dec 2020 00:59:43 -0800 Subject: [PATCH] Extend unranked hlo transformations to also support and, or and xor. PiperOrigin-RevId: 346270393 --- lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc index 1788b28..5ba5fd9 100644 --- a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc +++ b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc @@ -42,11 +42,11 @@ namespace { sep fn(SqrtOp) sep fn(TanhOp) // TODO(herhut): Generate these out of op definitions. -#define MAP_XLA_OPERATION_CWISE_BINARY(fn, sep) \ - fn(AddOp) sep fn(Atan2Op) sep fn(ComplexOp) sep fn(DivOp) sep fn(MaxOp) \ - sep fn(MinOp) sep fn(MulOp) sep fn(PowOp) sep fn(RemOp) \ - sep fn(ShiftLeftOp) sep fn(ShiftRightArithmeticOp) \ - sep fn(ShiftRightLogicalOp) sep fn(SubOp) +#define MAP_XLA_OPERATION_CWISE_BINARY(fn, sep) \ + fn(AddOp) sep fn(AndOp) sep fn(Atan2Op) sep fn(ComplexOp) sep fn(DivOp) \ + sep fn(MaxOp) sep fn(MinOp) sep fn(MulOp) sep fn(OrOp) sep fn(PowOp) \ + sep fn(RemOp) sep fn(ShiftLeftOp) sep fn(ShiftRightArithmeticOp) \ + sep fn(ShiftRightLogicalOp) sep fn(SubOp) sep fn(XorOp) // TODO(herhut): Generate these out of op definitions. #define MAP_CHLO_OPERATION_CWISE_UNARY(fn, sep) \