Searched refs:expanded_start_indices_shape (Results 1 – 1 of 1) sorted by relevance
3332 std::vector<int64> expanded_start_indices_shape; in InferGatherShape() local3335 expanded_start_indices_shape.reserve(start_indices_shape.dimensions_size()); in InferGatherShape()3339 std::back_inserter(expanded_start_indices_shape)); in InferGatherShape()3343 if (expanded_start_indices_shape.size() == in InferGatherShape()3345 expanded_start_indices_shape.push_back(1); in InferGatherShape()3350 input_shape, expanded_start_indices_shape, gather_dim_numbers)); in InferGatherShape()3392 (expanded_start_indices_shape.size() - 1); in InferGatherShape()3432 current_bound = expanded_start_indices_shape[gather_dims_seen++]; in InferGatherShape()