Support higher axis numbers for dr.any and dr.all reductions

[zhestyatsky]

Hello,

I have been experimenting with the latest drjit version and noticed that dr.all() and dr.any() operations seem to not be supported for Tensors when they are evaluated with axis argument greater than 0.

Reproducer below raises RuntimeError: drjit.all(<drjit.cuda.TensorXb>): tensor type is not compatible with the requested reduction.

import drjit as dr
from drjit.cuda import TensorXb

v = TensorXb([[True, False, True],
              [True, True, True],
              [True, False, True],
              [True, True, True]
             ])
dr.all(v, axis=1)

Similar exception is raised when calling dr.any().

This PR aims to enable dr.all() and dr.any() operations for Tensors with axis parameter greater than 0.