from torch.nn.modules.module import Module

from ..functions.roi_crop import RoICropFunction


class _RoICrop(Module):
    def __init__(self, layout = 'BHWD'):
        super(_RoICrop, self).__init__()
    def forward(self, input1, input2):
        return RoICropFunction()(input1, input2)
