Searched defs:permute_hidden (Results 1 – 4 of 4) sorted by relevance
283 def permute_hidden(self, hx: Tensor, permutation: Optional[Tensor]) -> Tensor: member in RNNBase623 def permute_hidden( # type: ignore[override] member in LSTM911 def permute_hidden(self, hx: Tensor, permutation: Optional[Tensor]) -> Tensor: member in GRU
366 def permute_hidden(self, hx: Tensor, permutation: Optional[Tensor]): member in RNNBase1013 def permute_hidden( # type: ignore[override] member in LSTM
355 Tensor RNNImplBase<Derived>::permute_hidden( in permute_hidden() function in torch::nn::detail::RNNImplBase595 std::tuple<Tensor, Tensor> LSTMImpl::permute_hidden( in permute_hidden() function in torch::nn::LSTMImpl
491 def permute_hidden( # type: ignore[override] member in LSTM