Home
last modified time | relevance | path

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

/external/pytorch/torch/nn/modules/
Drnn.py316 def get_expected_hidden_size( member in RNNBase
/external/pytorch/torch/csrc/api/src/nn/modules/
Drnn.cpp308 get_expected_hidden_size(const Tensor& input, const Tensor& batch_sizes) in get_expected_hidden_size() function in torch::nn::detail::RNNImplBase
/external/pytorch/torch/ao/nn/quantized/dynamic/modules/
Drnn.py250 def get_expected_hidden_size( member in RNNBase