Searched refs:sparse_tensor_indices (Results 1 – 2 of 2) sorted by relevance
461 bucket.sparse_tensor_indices = in mark_variable_ready_sparse()464 bucket.sparse_tensor_indices.value(), grad.indices(), false, false); in mark_variable_ready_sparse()986 bucket.sparse_tensor_indices); in all_reduce_bucket()1628 bucket.sparse_tensor_indices.value().numel() == in finalize_backward()1633 bucket.sparse_tensor_indices.value(), in finalize_backward()
392 std::optional<at::Tensor> sparse_tensor_indices = std::nullopt; member