def METHOD_NAME(input: dace.float32[2, 2], output: dace.float32[2, 2]):
    tmp_max = np.max(input, axis=axis)
    out_tmp = dace.define_local(out_tmp_shape, out_tmp_dtype)
    exp_minus_max(tmp_max=tmp_max, original_input=input, output=out_tmp)
    tmp_sum = np.sum(out_tmp, axis=axis)
    out_tmp_div_sum(out_tmp=out_tmp, tmp_sum=tmp_sum, output=output)