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

Method test_debug

caffe2/python/nomnigraph_test.py:146–153  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

144 assert fc.outputs[0].consumers[0].name == "Relu"
145
146 def test_debug(self):
147 nn = ng.NNModule()
148 dfg = nn.dataFlow
149 dfg.createNode(ng.NeuralNetData("X"))
150 dfg.createNode(ng.NeuralNetData("W"))
151 dfg.createNode(ng.NeuralNetOperator("Op"))
152
153 ng.render(nn.dataFlow)
154
155 def test_match_graph_node(self):
156 mg = ng.NNMatchGraph()

Callers

nothing calls this directly

Calls 2

createNodeMethod · 0.80
renderMethod · 0.45

Tested by

no test coverage detected