Searched refs:chunked_hgates (Results 1 – 2 of 2) sorted by relevance
781 auto chunked_hgates = params.linear_hh(hidden).unsafe_chunk(3, 1); in operator ()() local783 chunked_hgates[0].add_(chunked_igates[0]).sigmoid_(); in operator ()()785 chunked_hgates[1].add_(chunked_igates[1]).sigmoid_(); in operator ()()787 chunked_igates[2].add(chunked_hgates[2].mul_(reset_gate)).tanh_(); in operator ()()
3527 chunked_hgates = F.linear(cur_hidden, hh_weight, hh_bias).chunk(3, 2)3528 reset_gate = (chunked_hgates[0] + chunked_igates[0]).sigmoid()3529 input_gate = (chunked_hgates[1] + chunked_igates[1]).sigmoid()3530 new_gate = (chunked_igates[2] + (chunked_hgates[2] * reset_gate)).tanh()3536 chunked_hgates = F.linear(cur_hidden, hh_weight, hh_bias).chunk(3, 1)3537 reset_gate = (chunked_hgates[0] + chunked_igates[0]).sigmoid()3538 input_gate = (chunked_hgates[1] + chunked_igates[1]).sigmoid()3539 new_gate = (chunked_igates[2] + (chunked_hgates[2] * reset_gate)).tanh()