Searched refs:needs_64bit_indexing_no_split (Results 1 – 1 of 1) sorted by relevance
395 bool needs_64bit_indexing_no_split(const at::Tensor& input, const at::Tensor& weight) const { in needs_64bit_indexing_no_split() function427 if (needs_64bit_indexing_no_split(input, weight)) { in use_cudnn()505 if (needs_64bit_indexing_no_split(input, weight)) { in use_miopen()590 if (needs_64bit_indexing_no_split(input, weight)) { in use_mps()