MCPcopy
hub / github.com/dmlc/dgl / csrmask

Function csrmask

python/dgl/backend/tensorflow/sparse.py:456–461  ·  view source on GitHub ↗
(gidxA, A_weights, gidxB)

Source from the content-addressed store, hash-verified

454
455
456def csrmask(gidxA, A_weights, gidxB):
457 @tf.custom_gradient
458 def _lambda(A_weights):
459 return csrmask_real(gidxA, A_weights, gidxB)
460
461 return _lambda(A_weights)

Callers

nothing calls this directly

Calls 1

_lambdaFunction · 0.85

Tested by

no test coverage detected