Searched refs:gather_dims_seen (Results 1 – 1 of 1) sorted by relevance
3394 int64 gather_dims_seen = 0; in InferGatherShape() local3425 if (gather_dims_seen == gather_dim_numbers.index_vector_dim()) { in InferGatherShape()3426 gather_dims_seen++; in InferGatherShape()3430 expanded_start_indices_shape_dynamic_dimensions[gather_dims_seen]; in InferGatherShape()3432 current_bound = expanded_start_indices_shape[gather_dims_seen++]; in InferGatherShape()