Commit 1f915586 authored by rusty1s's avatar rusty1s
Browse files

typo

parent 65abb5e3
...@@ -45,9 +45,12 @@ def spline_weighting_backward_input(grad_output, weight, basis, ...@@ -45,9 +45,12 @@ def spline_weighting_backward_input(grad_output, weight, basis,
weight_index): # pragma: no cover weight_index): # pragma: no cover
grad_input = grad_output.new(grad_output.size(0), weight.size(1)) grad_input = grad_output.new(grad_output.size(0), weight.size(1))
func = get_func('weighting_backward_input', grad_output) func = get_func('weighting_backward_input', grad_output)
# Transpose for coalesced memory access.
weight = weight.transpose(1, 2).contiguous() weight = weight.transpose(1, 2).contiguous()
func(grad_input, grad_output, weight, basis, weight_index) func(grad_input, grad_output, weight, basis, weight_index)
weight = weight.transpose(1, 2).contiguous() weight = weight.transpose(1, 2).contiguous()
return grad_input return grad_input
......
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