MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / _check_shape

Function _check_shape

imperative/python/megengine/xla/rules/utils.py:68–70  ·  view source on GitHub ↗
(actual, ref)

Source from the content-addressed store, hash-verified

66
67
68def _check_shape(actual, ref):
69 if ref is not None:
70 assert _shape_equal(actual, ref), f"shape error, actual: {actual}, ref: {ref}"
71
72
73def _check_dtype(actual, ref):

Callers 5

_reduce_nokeepdimFunction · 0.85
_reduceFunction · 0.85
__init__Method · 0.85
get_var_shape_lowerFunction · 0.85
fill_lowerFunction · 0.85

Calls 1

_shape_equalFunction · 0.85

Tested by

no test coverage detected