Searched defs:check_forward_input (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ | ||
| H A D | rnn.cpp | 862 void RNNCellImplBase<Derived>::check_forward_input( in check_forward_input() function in torch::nn::detail::RNNCellImplBase |
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ | ||
| H A D | rnn.py | 1014 def check_forward_input(self, input): member in RNNCellBase |