Method main

tests/python/relax/test_transform_static_plan_block_memory.py:1688–1697

()

        @R.function
        def main():
            cls = Before
            x = R.builtin.alloc_tensor(R.shape([16, 16]), dtype="float32", runtime_device_index=0)
            x1 = R.memory.view(x, [128], "float32", 0)
            x2 = R.memory.ensure_zero_offset(x1)
            y = R.builtin.alloc_tensor(R.shape([128]), dtype="float32", runtime_device_index=0)
            cls.tir_exp(x2, y)
            z = R.builtin.alloc_tensor(R.shape([128]), dtype="float32", runtime_device_index=0)
            cls.tir_exp(y, z)
            return z

    @I.ir_module
    class Expected: