Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
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
68
def
_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
73
def
_check_dtype(actual, ref):
Callers
5
_reduce_nokeepdim
Function · 0.85
_reduce
Function · 0.85
__init__
Method · 0.85
get_var_shape_lower
Function · 0.85
fill_lower
Function · 0.85
Calls
1
_shape_equal
Function · 0.85
Tested by
no test coverage detected