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

Add shape inference function for XlaScatter

Browse Source 
PiperOrigin-RevId: 334603580
Change-Id: Idc193c32dd429cdf7f14a8496c6340c4e7b803b4
This commit is contained in:
A. Unique TensorFlower 2020-09-30 08:26:55 -07:00 committed by TensorFlower Gardener
parent 1c9964097d
commit fa75523767
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

@ -762,7 +762,7 @@ REGISTER_OP("XlaScatter")
.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::UnchangedShape)
.Doc(R"doc( .Doc(R"doc(
Wraps the XLA Scatter operator documented at Wraps the XLA Scatter operator documented at
https://www.tensorflow.org/xla/operation_semantics#scatter. https://www.tensorflow.org/xla/operation_semantics#scatter.