diff options
| author |  Jacques Pienaar <jpienaar@google.com> | 2018-04-18 21:15:41 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-18 21:18:35 -0700 |
| commit | 1d003ee5f82d4d044323a3f162e6cfcf6d645346 (patch) | |
| tree | 6cc32a695c30f8dbdfae4841e43ab8aeb8659728 /tensorflow/compiler/xla/service/shape_inference.cc | |
| parent | fd10bfb61ef6b1885c8fa2459522fa98305df703 (diff) | |
Initial addition of CLZ HLO
* Adds the HLO op and lowering on CPU/GPU/evaluator;
* This does not update the operation semantics;
PiperOrigin-RevId: 193461989
Diffstat (limited to 'tensorflow/compiler/xla/service/shape_inference.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/shape_inference.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/shape_inference.cc b/tensorflow/compiler/xla/service/shape_inference.cc index 77e12d3602..48b2922e77 100644 --- a/tensorflow/compiler/xla/service/shape_inference.cc +++ b/tensorflow/compiler/xla/service/shape_inference.cc @@ -52,6 +52,8 @@ UnaryOperation OpcodeToUnaryOperation(HloOpcode opcode) { return UNOP_ABS; case HloOpcode::kCeil: return UNOP_CEIL; + case HloOpcode::kClz: + return UNOP_CLZ; case HloOpcode::kCos: return UNOP_COS; case HloOpcode::kExp: @@ -360,6 +362,7 @@ StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, arg, primitive_util::ComplexComponentType(arg.element_type())); } return arg; + case UNOP_CLZ: case UNOP_NEGATE: case UNOP_ROUND_NEAREST_AFZ: case UNOP_SIGN: |