    def _infer_shapes(self, a: Tensor, b: Tensor, bias: Tensor):
        """Infers output shapes for gemm_rcr_bas.

        Parameters
        ----------
        a : Tensor
            Input tensor A.
        b : Tensor
            Input tensor B.
        bias : Tensor
            Input tensor bias. Must be a 1D vector.

        Returns
        -------
        List[IntVar]
            Output tensor shape.
        """
        return super()._infer_shapes(a, b)
Infers output shapes for gemm_rcr_bas.

Parameters
----------
a : Tensor
Input tensor A.
b : Tensor
Input tensor B.
bias : Tensor
Input tensor bias. Must be a 1D vector.

Returns
-------
List[IntVar]
Output tensor shape.
python/aitemplate/compiler/ops/gemm_epilogue_vistor/dual_gemm_rcr_silu.py