Home
last modified time | relevance | path

Searched refs:index_dims_in_start_indices (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc66 const int64 index_dims_in_start_indices = indices_are_scalar ? 0 : 1; in CanonicalizeGatherIndices() local
73 if (shape.dimensions_size() == index_dims_in_start_indices) { in CanonicalizeGatherIndices()
80 shape.dimensions_size() - index_dims_in_start_indices); in CanonicalizeGatherIndices()