    def __call__(self, x: Tensor) -> Tensor:
        """call the op

        Parameters
        ----------
        x : Tensor
            input tensor
        Returns
        ----------
            Tensor
        """
        self._attrs["inputs"] = [x]
        self._set_depth()
        output_shape = self._infer_shapes(x)
        self._extract_exec_path(x)
        output = Tensor(output_shape, src_ops={self}, dtype="int64")
        self._attrs["outputs"] = [output]
        return output
call the op

Parameters
----------
x : Tensor
input tensor
Returns
----------
Tensor
python/aitemplate/compiler/ops/tensor/argmax.py