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

Method test_relu_graph_simple

caffe2/python/trt/test_trt.py:109–111  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

107
108 @unittest.skipIf(not workspace.C.use_trt, "No TensortRT support")
109 def test_relu_graph_simple(self):
110 X = np.random.randn(1, 1, 3, 2).astype(np.float32)
111 self._test_relu_graph(X, 1, 50)
112
113
114 @unittest.skipIf(not workspace.C.use_trt, "No TensortRT support")

Callers

nothing calls this directly

Calls 3

_test_relu_graphMethod · 0.95
astypeMethod · 0.80
randnMethod · 0.45

Tested by

no test coverage detected