[mlir][sparse] Custom reduce with identity
Implement the new sparse_tensor.reduce operation which accepts a starting identity value and a code block describing how to perform the reduction. Reviewed by: aartbik Differential Revision: https://reviews.llvm.org/D130573
Loading
Please sign in to comment