Searched defs:weight_ih (Results 1 – 3 of 3) sorted by relevance
274 Tensor weight_ih; variable
248 auto weight_ih = _shuffle_weight(w0, rnn.mode); in mkldnn_rnn_layer() local338 auto weight_ih = _shuffle_weight(weight0, rnn.mode); in mkldnn_rnn_layer_backward() local
388 const at::Tensor& weight_ih, in get_lstm_packed_weights()