Home
last modified time | relevance | path

Searched defs:weight_hh (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/csrc/api/include/torch/nn/modules/
Drnn.h275 Tensor weight_hh; variable
/external/pytorch/aten/src/ATen/native/mkldnn/
DRNN.cpp249 auto weight_hh = _shuffle_weight(w1, rnn.mode); in mkldnn_rnn_layer() local
339 auto weight_hh = _shuffle_weight(weight1, rnn.mode); in mkldnn_rnn_layer_backward() local
DMKLDNNConversions.cpp389 const at::Tensor& weight_hh, in get_lstm_packed_weights()