Commit d3cb8b31 authored by rusty1s's avatar rusty1s
Browse files

typos

parent 76f6c267
......@@ -36,6 +36,7 @@ scatter_fw(torch::Tensor src, torch::Tensor index, int64_t dim,
return scatter_cpu(src, index, dim, optional_out, dim_size, reduce);
}
}
using torch::autograd::AutogradContext;
using torch::autograd::Variable;
using torch::autograd::variable_list;
......
......@@ -69,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover
f'matches your PyTorch install.')
from .scatter import (scatter_sum, scatter_add, scatter_mean, scatter_min,
scatter_max, scatter) # noqa: E402
scatter_max, scatter) # noqa
from .segment_csr import (segment_sum_csr, segment_add_csr, segment_mean_csr,
segment_min_csr, segment_max_csr, segment_csr,
gather_csr) # noqa: E402
gather_csr) # noqa
from .segment_coo import (segment_sum_coo, segment_add_coo, segment_mean_coo,
segment_min_coo, segment_max_coo, segment_coo,
gather_coo) # noqa: E402
gather_coo) # noqa
from .composite import (scatter_std, scatter_logsumexp, scatter_softmax,
scatter_log_softmax) # noqa: E402
scatter_log_softmax) # noqa
__all__ = [
'scatter_sum',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment