MCPcopy Create free account
hub / github.com/apache/tvm / get_ref

Function get_ref

tests/python/relax/test_codegen_dnnl.py:80–86  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

78
79 @memoize("relax.tests.test_codegen_dnnl.conv2d_relu_x2")
80 def get_ref():
81 data_np = np.random.randn(1, 64, 56, 56).astype("float32")
82 weight1_np = np.random.randn(64, 64, 3, 3).astype("float32")
83 weight2_np = np.random.randn(64, 64, 3, 3).astype("float32")
84 inputs = [data_np, weight1_np, weight2_np]
85 ref = build_and_run(Conv2dReLUx2, inputs, legalize=True)
86 return inputs, ref
87
88 inputs, ref = get_ref()
89

Callers 1

test_dnnl_offloadFunction · 0.70

Calls 2

build_and_runFunction · 0.70
astypeMethod · 0.45

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…