def init(self, N, C, H, W, device): self.inputs = {"input": torch.rand(N, C, H, W).to(device)} self.op = tq.FakeQuantize().to(device) self.set_module_name('FakeQuantize')
def init(self, N, C, H, W): self.input = torch.rand(N, C, H, W) self.op = tq.FakeQuantize() self.set_module_name('FakeQuantize')
def init(self, C, M, N): self.input = torch.rand(C, M, N) self.op = tq.FakeQuantize() self.set_module_name('FakeQuantize')