Searched refs:check_rnn_cell_forward_input (Results 1 – 1 of 1) sorted by relevance
676 void check_rnn_cell_forward_input(const Tensor& input, const c10::SymInt& input_size) { in check_rnn_cell_forward_input() function1535 check_rnn_cell_forward_input(input, w_ih.sym_size(1)); in lstm_cell()1648 check_rnn_cell_forward_input(input, w_ih.size(1)); in gru_cell()1663 check_rnn_cell_forward_input(input, w_ih.size(1)); in rnn_tanh_cell()1677 check_rnn_cell_forward_input(input, w_ih.size(1)); in rnn_relu_cell()