Change sparse.reduce_max(sp_input, axis, keepdims, reduction_axes, keep_dims)
to sparse.reduce_max(sp_input, axis, keepdims, output_is_sparse, name) and unify with reduce_max_sparse via dispatching (see new output_is_sparse_arg). PiperOrigin-RevId: 221879420
Loading
Please sign in to comment