nki.isa.tensor_partition_reduce#

nki.isa.tensor_partition_reduce(dst, op, data, name=None)[source]#

Apply a reduction operation across partitions of an input data tile using GpSimd Engine.

Parameters:

  • dst – output tile with reduced result

  • op – the reduction operator (add, max, bitwise_or, bitwise_and)

  • data – the input tile to be reduced