diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-09-26 07:31:25 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-09-26 07:35:27 -0700 |
commit | 36649e842908d89a3dc44a840bd6305fe401123f (patch) | |
tree | 3a58307b3995fd1590634ecb382b8f644717189f /tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc | |
parent | 818dc210881aadcce9013b693e071b96a6e1f1f6 (diff) |
Adds XLA support for GatherV2 (gather with axis parameter).
PiperOrigin-RevId: 170050380
Diffstat (limited to 'tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc')
-rw-r--r-- | tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc b/tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc index c42d8b97ea..e2d3d40813 100644 --- a/tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc +++ b/tensorflow/compiler/tf2xla/kernels/tensor_array_ops.cc @@ -311,7 +311,7 @@ class TensorArrayGatherOp : public XlaOpKernel { xla::ComputationDataHandle ta = resource->value; xla::ComputationDataHandle gather = XlaComputeGatherDynamicSlice( - ctx, ta, ta_shape, indices, indices_shape, dtype_, b); + ctx, ta, ta_shape, indices, indices_shape, 0, dtype_, b); ctx->SetOutput(0, gather); } |