experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add additional elementwise unary ops to hlo_shape_derivation.

Browse Source 
This makes them work with dynamic shapes.

PiperOrigin-RevId: 299311603
Change-Id: Id3801de8406ffc0054a7f6cf6c3ac25c4dd95f55
This commit is contained in:
Adrian Kuegel 2020-03-06 02:34:32 -08:00 committed by TensorFlower Gardener
parent df9f9fb7f3
commit 15e945a791
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/mlir/xla/transforms/hlo_shape_derivation.h
View File

@ -114,11 +114,16 @@ SAME_SHAPE_AS_FIRST_OPERAND(CeilOp)
SAME_SHAPE_AS_FIRST_OPERAND(CosOp)
SAME_SHAPE_AS_FIRST_OPERAND(DivOp)
SAME_SHAPE_AS_FIRST_OPERAND(ExpOp)
SAME_SHAPE_AS_FIRST_OPERAND(LogOp)
SAME_SHAPE_AS_FIRST_OPERAND(MaxOp)
SAME_SHAPE_AS_FIRST_OPERAND(MinOp)
SAME_SHAPE_AS_FIRST_OPERAND(MulOp)
SAME_SHAPE_AS_FIRST_OPERAND(NegOp)
SAME_SHAPE_AS_FIRST_OPERAND(RemOp)
SAME_SHAPE_AS_FIRST_OPERAND(RsqrtOp)
SAME_SHAPE_AS_FIRST_OPERAND(SignOp)
SAME_SHAPE_AS_FIRST_OPERAND(SinOp)
SAME_SHAPE_AS_FIRST_OPERAND(SqrtOp)
SAME_SHAPE_AS_FIRST_OPERAND(SubOp)
SAME_SHAPE_AS_FIRST_OPERAND(TanhOp)