ca776ef5de4dae63c328235379f261a05dc47748,sru/cuda_functional.py,SRU_Compute_GPU,backward,#SRU_Compute_GPU#,97

Before Change

        //grad_wc = x.new(2*bidir*d).zero_()
        //grad_bias = x.new(2*bidir*d).zero_()
        grad_wc = x.new(2, batch, bidir*d).zero_()
        grad_bias = x.new(2, batch, bidir*d).zero_()
        grad_init = x.new(batch, d*bidir)
        grad_x = x.new(*x.size()).zero_() if skip_type > 0 and k_ == 3 else None

        if skip_type > 0 and k_ == 3:

After Change

        //grad_wc = x.new(2*bidir*d).zero_()
        //grad_bias = x.new(2*bidir*d).zero_()
        grad_wc = x.new_zeros(2, batch, bidir*d)
        grad_bias = x.new_zeros(2, batch, bidir*d)
        grad_init = x.new_zeros(batch, d*bidir)
        grad_x = x.new_zeros(*x.size()) if skip_type > 0 and k_ == 3 else None

        if skip_type > 0 and k_ == 3:

Instances