Searched defs:cudnn_input_size (Results 1 – 1 of 1) sorted by relevance
1706 auto cudnn_input_size = input_r.size(-1); in _cudnn_rnn_backward_input() local1935 auto cudnn_input_size = input_r.size(-1); in _cudnn_rnn_backward_weight() local2415 auto cudnn_input_size = input.size(-1); in try_get_weight_buf() local