MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / forward

Method forward

imperative/python/test/unit/module/test_qat.py:51–55  ·  view source on GitHub ↗
(self, inp)

Source from the content-addressed store, hash-verified

49 )
50
51 def forward(self, inp):
52 out = self.quant(inp)
53 out = self.conv_bn(out)
54 out = self.dequant(out)
55 return out
56
57 inputs = tensor(np.random.randn(4, in_channels, 32, 32).astype(np.float32))
58 for groups, bias in product([1, 4], [True, False]):

Callers

nothing calls this directly

Calls 1

expand_dimsFunction · 0.90

Tested by

no test coverage detected