to simplify flt and partion layer implementation This reverts commit 2e49e1bc5c6a316e407c430596013fa3a5c13f84.