Function _make_identity_map

tests/python/test_out_channel.py:9–13

Create a MapSpec that returns the input buffer unchanged.

(buf: Buffer)

def _make_identity_map(buf: Buffer):
    """Create a MapSpec that returns the input buffer unchanged."""
    x = kernel.arg("x")
    spec = kernel.elementwise(x)
    return kernel.map(spec, x=buf)


class TestGoWithOutChannel: