Searched defs:hiddens (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 651 Tensor hidden_concat(at::ArrayRef<Tensor> hiddens) { return at::cat(hiddens, 0); } in hidden_concat()
652 tpair_of<Tensor> hidden_concat(at::ArrayRef<tpair_of<Tensor>> hiddens) { in hidden_concat()
933 std::vector<hidden_type> hiddens; in operator ()() local
1081 const std::vector<hidden_type>& hiddens, const std::vector<weight_type>& weights, in apply_layer_stack()
1111 const std::vector<typename CellType::hidden_type>& hiddens, in _rnn_impl()
1128 const std::vector<typename CellType::hidden_type>& hiddens, in _rnn_impl_with_concat()
1144 std::vector<typename LSTMCell<cell_params>::hidden_type> hiddens; in _lstm_impl() local
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 2967 def params_hiddens(params, hiddens, i, bidirectional): argument
2978 def update_hidden_for_packed(cur_hidden, last_batch_size, batch_size, hiddens): argument
|
| /external/pytorch/test/ |
| D | test_jit.py | 2522 def forward(self, x, lengths, hiddens): argument
|