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')
