Make boundary restrictor batched