@functools.partial(bool_to_int8, argnums=0)
def _gather(operand, start_indices, *, dimension_numbers, slice_sizes, _in_avals, _out_aval):
    'Tensorflow implementation of gather.'
    del _in_avals
    if (not _enable_xla):
        raise _xla_disabled_error('gather')
    proto = _gather_dimensions_proto(start_indices.shape, dimension_numbers)
    slice_sizes_tf = _eval_shape(slice_sizes)
    out = tfxla.gather(operand, start_indices, proto, slice_sizes_tf, False)
    out.set_shape(_aval_to_tf_shape(_out_aval))
    return out
Tensorflow implementation of gather.
@functools.partial(bool_to_int8, argnums=0)
def _gather(operand, start_indices, *, dimension_numbers, slice_sizes, _in_avals, _out_aval):
    
    del _in_avals
    if (not _enable_xla):
        raise _xla_disabled_error('gather')
    proto = _gather_dimensions_proto(start_indices.shape, dimension_numbers)
    slice_sizes_tf = _eval_shape(slice_sizes)
    out = tfxla.gather(operand, start_indices, proto, slice_sizes_tf, False)
    out.set_shape(_aval_to_tf_shape(_out_aval))
    return out