def conv2d_gen_function(
    func_attrs,
    exec_cond_template,
    shape_eval_template,
    shape_save_template,
):
    """Generates function body.

    Parameters
    ----------
    func_attrs : Dict
        Operation attributes.
    exec_cond_template : jinja2.Template
        Generates if statement to execute kernel.
    shape_eval_template : jinja2.Template
        Generates shape calculation.
        The template is passed from compiler/ops/pool.
    shape_save_template : jinja2.Template
        Generates output dimensions.
        The template is passed from compiler/ops/pool.

    Returns
    -------
    str
        The rendered template of generated function body.
    """
    return common.gen_function(
        func_attrs,
        exec_cond_template,
        shape_eval_template,
        shape_save_template,
        "bias_relu",
        extra_code=EXTRA_CODE.render(),
    )
Generates function body.

Parameters
----------
func_attrs : Dict
Operation attributes.
exec_cond_template : jinja2.Template
Generates if statement to execute kernel.
shape_eval_template : jinja2.Template
Generates shape calculation.
The template is passed from compiler/ops/pool.
shape_save_template : jinja2.Template
Generates output dimensions.
The template is passed from compiler/ops/pool.

Returns
-------
str
The rendered template of generated function body.
python/aitemplate/backend/rocm/conv2d/transposed_conv2d_bias_relu.py