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, ...@@ -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); return scatter_cpu(src, index, dim, optional_out, dim_size, reduce);
} }
} }
using torch::autograd::AutogradContext; using torch::autograd::AutogradContext;
using torch::autograd::Variable; using torch::autograd::Variable;
using torch::autograd::variable_list; using torch::autograd::variable_list;
......
...@@ -69,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover ...@@ -69,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover
f'matches your PyTorch install.') f'matches your PyTorch install.')
from .scatter import (scatter_sum, scatter_add, scatter_mean, scatter_min, 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, from .segment_csr import (segment_sum_csr, segment_add_csr, segment_mean_csr,
segment_min_csr, segment_max_csr, segment_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, from .segment_coo import (segment_sum_coo, segment_add_coo, segment_mean_coo,
segment_min_coo, segment_max_coo, segment_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, from .composite import (scatter_std, scatter_logsumexp, scatter_softmax,
scatter_log_softmax) # noqa: E402 scatter_log_softmax) # noqa
__all__ = [ __all__ = [
'scatter_sum', '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