get_expected_cell_size (definition) in projects: aosp_15_r20

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

/aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/
H A D	rnn.py	`502 def get_expected_cell_size( member in LSTM`
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A D	rnn.cpp	`564 std::tuple<int64_t, int64_t, int64_t> LSTMImpl::get_expected_cell_size( in get_expected_cell_size() function in torch::nn::LSTMImpl`
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A D	rnn.py	`977 def get_expected_cell_size( member in LSTM`