projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 4faee39) | patch
Generalize the gather_indices dimension that stores indices
authorSanjoy Das <sanjoy@google.com>
Mon, 26 Feb 2018 18:17:15 +0000 (10:17 -0800)
committerGunhan Gulsoy <gunan@google.com>
Tue, 27 Feb 2018 22:33:33 +0000 (14:33 -0800)
commit0a799feaea50d4e48e8daa1f3954427fdccd76f1
treeb879159f4d9200edf8456aa515cb5a056e768315tree | snapshot
parent4faee3942d9983e0c96091b32095cc0d9ff494e0commit | diff
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
tensorflow/compiler/xla/service/hlo_instruction.cc diff | blob | history
tensorflow/compiler/xla/service/hlo_instruction.h diff | blob | history
tensorflow/compiler/xla/service/hlo_instruction_test.cc diff | blob | history
tensorflow/compiler/xla/service/shape_inference.cc diff | blob | history
tensorflow/compiler/xla/service/shape_inference_test.cc diff | blob | history
tensorflow/compiler/xla/xla_data.proto diff | blob | history
tensorflow/docs_src/performance/xla/operation_semantics.md diff | blob | history
Domain: Machine Learning / ML Framework;