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

Function _check

tests/python/relax/test_transform_fuse_ops.py:28–32  ·  view source on GitHub ↗
(mod_actual, mod_expected)

Source from the content-addressed store, hash-verified

26
27
28def _check(mod_actual, mod_expected):
29 mod_actual = relax.transform.AnnotateTIROpPattern()(mod_actual)
30 mod_actual = relax.transform.FuseOps()(mod_actual)
31 mod_expected = relax.transform.AnnotateTIROpPattern()(mod_expected)
32 tvm.ir.assert_structural_equal(mod_actual, mod_expected)
33
34
35def test_fuse_simple():

Callers 15

test_fuse_simpleFunction · 0.70
test_conv2d_fuseFunction · 0.70
test_concatenateFunction · 0.70
test_tuple_rootFunction · 0.70
test_fuse_tuple_get_elemwiseFunction · 0.70
test_tuple_get_rootFunction · 0.70
test_tuple_intermediateFunction · 0.70
test_tuple_consecutiveFunction · 0.70
test_inception_likeFunction · 0.70
test_fuse_parallel_injectiveFunction · 0.70
test_softmaxFunction · 0.70
test_multiple_relax_functionsFunction · 0.70

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…