triton.experimental.gluon.language.reduce_or

triton.experimental.gluon.language.reduce_or(input, axis, keep_dims=False)

Returns the reduce_or of all elements in the input tensor along the provided axis

The reduction operation should be associative and commutative.

Parameters:
  • input (Tensor) – the input values

  • axis (int) – the dimension along which the reduction should be done. If None, reduce all dimensions

  • keep_dims (bool) – if true, keep the reduced dimensions with length 1

This function can also be called as a member function on tensor, as x.reduce_or(...) instead of reduce_or(x, ...).