def _reduce_window(operand, init_value, *, jaxpr, consts, window_dimensions, window_strides, padding, base_dilation, window_dilation, _in_avals, _out_aval):
    'TensorFlow implementation of reduce_window.\n\n  Args:\n    operand: N dimensional array containing elements of type T\n    init_value: starting value of the reduction\n    jaxpr: the jaxpr corresponding to the reduction function\n    consts: the constants associated with jaxpr.\n    window_dimensions: array of integers for window dimension values\n    window_strides: array of integers for window stride values\n    padding: array of pairs of integers for padding values\n    base_dilation: array of integers for base dilation values\n    window_dilation: array of integers for window dilation values\n\n  Returns:\n    The reduced operand.\n  '
    assert (len(consts) == 0), 'Reduction computation cannot have constants'
    if (not _enable_xla):
        raise _xla_disabled_error('reduce_window')

    def reducer(arg1: TfVal, arg2: TfVal) -> TfVal:
        closed_jaxpr = core.ClosedJaxpr(jaxpr, consts)
        (res,) = _interpret_jaxpr(closed_jaxpr, arg1, arg2)
        return res
    return _common_reduce_window(operand, init_value, reducer, window_dimensions, window_strides, padding, base_dilation, window_dilation, _in_avals, _out_aval)
TensorFlow implementation of reduce_window.

Args:
  operand: N dimensional array containing elements of type T
  init_value: starting value of the reduction
  jaxpr: the jaxpr corresponding to the reduction function
  consts: the constants associated with jaxpr.
  window_dimensions: array of integers for window dimension values
  window_strides: array of integers for window stride values
  padding: array of pairs of integers for padding values
  base_dilation: array of integers for base dilation values
  window_dilation: array of integers for window dilation values

Returns:
  The reduced operand.
def _reduce_window(operand, init_value, *, jaxpr, consts, window_dimensions, window_strides, padding, base_dilation, window_dilation, _in_avals, _out_aval):
    'TensorFlow implementation of reduce_window.\n\n  Args:\n    operand: N dimensional array containing elements of type T\n    init_value: starting value of the reduction\n    jaxpr: the jaxpr corresponding to the reduction function\n    consts: the constants associated with jaxpr.\n    window_dimensions: array of integers for window dimension values\n    window_strides: array of integers for window stride values\n    padding: array of pairs of integers for padding values\n    base_dilation: array of integers for base dilation values\n    window_dilation: array of integers for window dilation values\n\n  Returns:\n    The reduced operand.\n  '
    assert (len(consts) == 0), 'Reduction computation cannot have constants'
    if (not _enable_xla):
        raise _xla_disabled_error('reduce_window')

    def reducer(arg1: TfVal, arg2: TfVal) -> TfVal:
        closed_jaxpr = core.ClosedJaxpr(jaxpr, consts)
        (res,) = _interpret_jaxpr(closed_jaxpr, arg1, arg2)
        return res
    return _common_reduce_window(operand, init_value, reducer, window_dimensions, window_strides, padding, base_dilation, window_dilation, _in_avals, _out_aval)