Commit 0a799fea authored by Sanjoy Das's avatar Sanjoy Das Committed by Gunhan Gulsoy
Browse files

Generalize the gather_indices dimension that stores indices 

This is now exposed as a index_vector_dim dimension number.

Also fixed an off-by-one error in ValidateGatherDimensionNumbers in the
expression computing output_shape_rank.

PiperOrigin-RevId: 187040748
parent 4faee394
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment