core flow: add types for more segmented ops