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

Function _check_inference

tests/python/relax/test_op_manipulate.py:53–55  ·  view source on GitHub ↗
(bb: relax.BlockBuilder, call: relax.Call, expected_sinfo: relax.StructInfo)

Source from the content-addressed store, hash-verified

51
52
53def _check_inference(bb: relax.BlockBuilder, call: relax.Call, expected_sinfo: relax.StructInfo):
54 ret = bb.normalize(call)
55 tvm.ir.assert_structural_equal(ret.struct_info, expected_sinfo)
56
57
58def test_reshape_infer_struct_info():

Callers 15

test_reshape_infer_struct_infoFunction · 0.70
test_reshape_infer_struct_info_shape_symbolicFunction · 0.70
test_reshape_infer_struct_info_shape_varFunction · 0.70
test_reshape_infer_struct_info_more_input_dtypeFunction · 0.70
test_permute_dims_infer_struct_infoFunction · 0.70
test_permute_dims_infer_struct_info_shape_symbolicFunction · 0.70
test_permute_dims_infer_struct_info_shape_varFunction · 0.70
test_permute_dims_infer_struct_info_more_input_dtypeFunction · 0.70
test_expand_dims_infer_struct_infoFunction · 0.70
test_expand_dims_infer_struct_info_shape_symbolicFunction · 0.70
test_expand_dims_infer_struct_info_shape_varFunction · 0.70
test_expand_dims_infer_struct_info_more_input_dtypeFunction · 0.70

Calls 1

normalizeMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…