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