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

NFC: Update Gather op comments to reflect support for batch_dims=rank(indices)

Browse Source 
PiperOrigin-RevId: 285255979
Change-Id: I39a3664a10c9400dc8440549033c6fa8192f22f2
This commit is contained in:
Smit Hinsu 2019-12-12 13:18:21 -08:00 committed by TensorFlower Gardener
parent f7f4049ded
commit 606a542c81
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2xla/kernels/gather_op.cc
View File

@ -257,7 +257,7 @@ class GatherOp : public XlaOpKernel {
TF_DISALLOW_COPY_AND_ASSIGN(GatherOp);
// The number of batch dimensions, as passed in the batch_dims attribute.
// It must be less than rank(indices).
// It must be less than or equal to rank(indices).
int32 batch_dims_ = 0;
};

2
tensorflow/core/kernels/gather_op.cc
View File

@ -182,7 +182,7 @@ class GatherOp : public OpKernel {
private:
// The number of batch dimensions, as passed in the batch_dims attribute.
// It must be less than rank(indices).
// It must be less than or equal to rank(indices).
int32 batch_dims_ = 0;
};