Searched defs:zero_ks (Results 1 – 3 of 3) sorted by relevance
161 …auto zero_ks = at::DispatchKeySet(c10::DispatchKey::MPS) | at::DispatchKeySet(c10::DispatchKey::Ze… in _efficientzerotensor_mps() local
74 …auto zero_ks = at::DispatchKeySet(c10::DispatchKey::CUDA) | at::DispatchKeySet(c10::DispatchKey::Z… in _efficientzerotensor_cuda() local
1329 …auto zero_ks = at::DispatchKeySet(c10::DispatchKey::CPU) | at::DispatchKeySet(c10::DispatchKey::Ze… in _efficientzerotensor() local1342 …auto zero_ks = at::DispatchKeySet(c10::DispatchKey::Meta) | at::DispatchKeySet(c10::DispatchKey::Z… in _efficientzerotensor_meta_symint() local