Function _lambda

python/dgl/backend/tensorflow/sparse.py:198–199

(X, Y)

def gspmm(gidx, op, reduce_op, X, Y):
    @tf.custom_gradient
    def _lambda(X, Y):
        return gspmm_real(gidx, op, reduce_op, X, Y)

    if X is None:
        X = tf.zeros(())