Searched refs:expanded_start_indices_shape (Results 1 – 1 of 1) sorted by relevance
2987 std::vector<int64> expanded_start_indices_shape; in InferGatherShape() local2988 expanded_start_indices_shape.reserve(start_indices_shape.dimensions_size()); in InferGatherShape()2990 std::back_inserter(expanded_start_indices_shape)); in InferGatherShape()2991 if (expanded_start_indices_shape.size() == in InferGatherShape()2993 expanded_start_indices_shape.push_back(1); in InferGatherShape()2997 input_shape, expanded_start_indices_shape, gather_dim_numbers)); in InferGatherShape()3039 (expanded_start_indices_shape.size() - 1); in InferGatherShape()3058 current_bound = expanded_start_indices_shape[gather_dims_seen++]; in InferGatherShape()