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

Method calc_conv

imperative/python/megengine/module/conv.py:920–933  ·  view source on GitHub ↗
(self, inp, weight, offset, mask, bias)

Source from the content-addressed store, hash-verified

918 return (1, self.out_channels, 1, 1)
919
920 def calc_conv(self, inp, weight, offset, mask, bias):
921 return deformable_conv2d(
922 inp,
923 weight,
924 offset,
925 mask,
926 bias,
927 self.stride,
928 self.padding,
929 self.dilation,
930 self.groups,
931 self.conv_mode,
932 self.compute_mode,
933 )
934
935 def forward(self, inp, offset, mask):
936 return self.calc_conv(inp, self.weight, offset, mask, self.bias)

Callers 1

forwardMethod · 0.95

Calls 1

deformable_conv2dFunction · 0.85

Tested by

no test coverage detected