MCPcopy Create free account
hub / github.com/pytorch/pytorch / _base_test_net

Method _base_test_net

caffe2/python/transformations_test.py:34–37  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

32
33class TestTransformations(tu.TestCase):
34 def _base_test_net(self):
35 net = core.Net("net")
36 net.Conv(["X", "w", "b"], ["Y"], stride=1, pad=0, kernel=3, order="NCHW")
37 return net
38
39 def _add_nnpack(self, net):
40 transformer.AddNNPACK(net)

Calls 2

NetMethod · 0.80
ConvMethod · 0.80

Tested by

no test coverage detected