Searched refs:gather_dims_seen (Results 1 – 1 of 1) sorted by relevance
3041 int64 gather_dims_seen = 0; in InferGatherShape() local3055 if (gather_dims_seen == gather_dim_numbers.index_vector_dim()) { in InferGatherShape()3056 gather_dims_seen++; in InferGatherShape()3058 current_bound = expanded_start_indices_shape[gather_dims_seen++]; in InferGatherShape()