↓ 1 callersFunctiongetSparseReshapePDLPatternsparse_ids = sparse_ops.sparse_reshape(sparse_ids, [ math_ops.reduce_prod( array_ops.slice(original_shape, [0], [original_rank - 1])), array_ops.gathe
tao_compiler/mlir/disc/transforms/disc_sparse_op_rewriter.cc:44