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

Function check_param

imperative/python/megengine/xla/rules/nn.py:373–376  ·  view source on GitHub ↗
(param, info)

Source from the content-addressed store, hash-verified

371 assert len(ishape) == len(oshape), f"shape error: {ishape} {oshape}"
372
373 def check_param(param, info):
374 assert len(ishape) == len(
375 param
376 ), f"pooling: illegal {info} {param} for {ishape}"
377
378 base_dilation = base_dilation if base_dilation is not None else (1, 1, 1, 1)
379 kernel_dilation = kernel_dilation if kernel_dilation is not None else (1, 1, 1, 1)

Callers 3

_poolingFunction · 0.85
scn_do_executeMethod · 0.85
check_execMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected