Searched refs:indices_shape_no_index_vectors (Results 1 – 1 of 1) sorted by relevance
68 TensorShape indices_shape_no_index_vectors{indices_shape}; in XlaGather() local70 indices_shape_no_index_vectors.RemoveLastDims(1); in XlaGather()75 out_shape.AppendShape(indices_shape_no_index_vectors); in XlaGather()