tep_size_setter_fn</th>
      <td><p>A function with the signature
<code>(kernel_results, new_step_size) -&gt; new_kernel_results</code> where
<code>kernel_results</code> are the results of the <code>inner_kernel</code>, <code>new_step_size</code>
is a <code>Tensor</code> or a nested collection of <code>Tensor</code>s with the same
structure as returned by the <code>step_size_getter_fn</code>, and
<code>new_kernel_results</code> are a copy of <code>kernel_results</code> with the step
size(s) set.</p></td>
 