Searched defs:chunked_gates (Results 1 – 3 of 3) sorted by relevance
247 auto chunked_gates = gates.chunk(4, 1); in lstm() local
747 auto chunked_gates = gates.unsafe_chunk(4, 1); in operator ()() local1566 auto chunked_gates = gates.unsafe_chunk(4, 1); in _thnn_differentiable_lstm_cell_backward() local
7097 auto chunked_gates = gate.unsafe_chunk(4, 1); in mkldnn_rnn_layer_differentiable_backward() local