* move segment_pool to phi.

* mark summed ids as optional tensor.

* fix as reviews.