Method invoke_self_multiple_times

axlearn/common/module_test.py:376–382

(self, x: Tensor, n: int)

        return x

    def invoke_self_multiple_times(self, x: Tensor, n: int) -> Tensor:
        # To call self.inc multiple times, create a context explicitly for each invocation
        # with a different `name`.
        for i in range(n):
            with child_context(f"self_call{i}", module=self):
                x = self.inc(x)
        return x

    def invoke_child_multiple_times_causing_conflict(self, x: Tensor, n: int) -> Tensor:
        for _ in range(n):