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

Fixes shape setting in tf2xla.gather

Browse Source 
PiperOrigin-RevId: 306877898
Change-Id: Icdaf04ef63d5cb3a3757d5b7514b76f4e4add44b
This commit is contained in:
A. Unique TensorFlower 2020-04-16 10:47:49 -07:00 committed by TensorFlower Gardener
parent ffb230a4b7
commit fcb0534cc5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2xla/ops/xla_ops.cc
View File

@ -674,7 +674,7 @@ REGISTER_OP("XlaGather")
.Attr("T: numbertype") .Attr("T: numbertype")
.Attr("Tindices: {int32, int64}") .Attr("Tindices: {int32, int64}")
.Output("output: T") .Output("output: T")
.SetShapeFn(UnchangedRank) .SetShapeFn(shape_inference::UnknownShape)
.Doc(R"doc( .Doc(R"doc(
Wraps the XLA Gather operator documented at Wraps the XLA Gather operator documented at
https://www.tensorflow.org/xla/operation_semantics#gather https://www.tensorflow.org/xla/operation_semantics#gather