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

Method __init__

imperative/python/test/unit/module/test_qat.py:43–49  ·  view source on GitHub ↗
(self, groups, bias)

Source from the content-addressed store, hash-verified

41
42 class TestNet(Module):
43 def __init__(self, groups, bias):
44 super().__init__()
45 self.quant = QuantStub()
46 self.dequant = DequantStub()
47 self.conv_bn = ConvBn2d(
48 in_channels, out_channels, kernel_size, groups=groups, bias=bias,
49 )
50
51 def forward(self, inp):
52 out = self.quant(inp)

Callers

nothing calls this directly

Calls 12

QuantStubClass · 0.90
DequantStubClass · 0.90
ConvBn2dClass · 0.90
ConvTransposeBn2dClass · 0.90
Conv2dClass · 0.90
ConvRelu2dClass · 0.90
ConvTranspose2dClass · 0.90
ConvTransposeRelu2dClass · 0.90
LinearBn1dClass · 0.90
LinearReluClass · 0.90
LinearBnRelu1dClass · 0.90

Tested by

no test coverage detected