Method setup

bench/python/all_ops/op_reformat.py:25–27

(self, input)

class OpReformatNCHWToNHWC(AbstractOpBase):
    def setup(self, input):
        super().setup(input)
        self.input_nchw = cvcuda.reformat(input, "NCHW")

    def run(self, input):
        return cvcuda.reformat(self.input_nchw, "NHWC")