Searched refs:_is_sparse_compressed_tensor (Results 1 – 1 of 1) sorted by relevance
33 def _is_sparse_compressed_tensor(obj: torch.Tensor): function43 return _is_sparse_compressed_tensor(obj) or obj.layout is torch.sparse_coo125 elif _is_sparse_compressed_tensor(x):356 if _is_sparse_compressed_tensor(entry):927 elif _is_sparse_compressed_tensor(inp):1676 elif _is_sparse_compressed_tensor(x):