Searched defs:proj_size (Results 1 – 5 of 5) sorted by relevance
30 int64_t proj_size, in _cudnn_rnn_cast_reflatten()
83 int64_t proj_size, in _cudnn_rnn_backward()154 int64_t proj_size; member1245 int64_t proj_size, in copy_weights_to_flat_buf_views()2101 int64_t proj_size, in _cudnn_rnn_backward()2389 c10::SymInt proj_size, in try_get_weight_buf()2511 SymInt proj_size = 0; in _cudnn_impl() local2585 c10::SymInt proj_size = 0; in _cudnn_impl() local
5926 proj_size, argument
1278 auto proj_size = hidden_size; in Create() local
11291 def check_lengths(lengths, enforce_sorted, use_default_hiddens, proj_size): argument