def METHOD_NAME(self, inputs, inputs_cl=None):
    if inputs_cl is None:
        inputs_cl = []
        for inp in inputs:
            b, c, h, w = inp.shape
            x = nn.Variable([b, h, w, c])
            x.d = inp.d.copy().transpose([0, 2, 3, 1])
            inputs_cl.append(x)
    self.inputs_cl = inputs_cl
    # Replace the very beginning of input
    for inp, inp_cl in zip(inputs, inputs_cl):
        f = inp.function_references[0]
        self.init_map_func_inputs(f, [inp_cl])