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

Function stack_lower

imperative/python/megengine/xla/rules/tensor.py:802–803  ·  view source on GitHub ↗
(ctx, *args: Union[HLOTensor, Sequence[HLOTensor]])

Source from the content-addressed store, hash-verified

800
801@register_lower_rule(mops.Stack)
802def stack_lower(ctx, *args: Union[HLOTensor, Sequence[HLOTensor]]):
803 return stack(args, ctx.op.axis)
804
805
806@register_lower_rule(mops.Linspace)

Callers

nothing calls this directly

Calls 1

stackFunction · 0.70

Tested by

no test coverage detected