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

Function set_symbolic_shape

imperative/python/megengine/utils/network.py:535–542  ·  view source on GitHub ↗

r"""Set the VarNode use symbolic shape or not, return the last status. Please set to True and must recover after dump if want to change the input batch size. Args: option: True for enable symbolic shape.

(option: bool)

Source from the content-addressed store, hash-verified

533
534
535def set_symbolic_shape(option: bool):
536 r"""Set the VarNode use symbolic shape or not, return the last status.
537 Please set to True and must recover after dump if want to change the input batch size.
538
539 Args:
540 option: True for enable symbolic shape.
541 """
542 return _set_symbolic_shape(option)
543
544
545def as_varnode(obj):

Callers 11

test_set_symbolic_shapeFunction · 0.90
test_split_basicFunction · 0.90
test_squeezeFunction · 0.90
__enter__Method · 0.50
__exit__Method · 0.50
trace_moduleFunction · 0.50
trace_normalMethod · 0.50
trace_without_hostMethod · 0.50
test_preluFunction · 0.50

Calls

no outgoing calls

Tested by 5

test_set_symbolic_shapeFunction · 0.72
test_split_basicFunction · 0.72
test_squeezeFunction · 0.72
test_preluFunction · 0.40