Searched defs:weight_hh (Results 1 – 3 of 3) sorted by relevance
275 Tensor weight_hh; variable
249 auto weight_hh = _shuffle_weight(w1, rnn.mode); in mkldnn_rnn_layer() local339 auto weight_hh = _shuffle_weight(weight1, rnn.mode); in mkldnn_rnn_layer_backward() local
389 const at::Tensor& weight_hh, in get_lstm_packed_weights()