Searched refs:sparse_csr_ks (Results 1 – 2 of 2) sorted by relevance
694 constexpr DispatchKeySet sparse_csr_ks = DispatchKeySet(DispatchKey::SparseCsr); variable
1074 return key_set_.has_all(c10::sparse_csr_ks); in is_sparse_compressed()1275 c10::sparse_ks | c10::sparse_csr_ks | c10::mkldnn_ks; in layout()