Files
dmlc--dgl/python/dgl/function/base.py
T
2026-07-13 13:35:51 +08:00

32 lines
604 B
Python

"""Built-in function base class"""
from __future__ import absolute_import
__all__ = ["BuiltinFunction", "TargetCode"]
class TargetCode(object):
"""Code for target
Note: must be consistent with the target code definition in C++ side:
src/kernel/binary_reduce_common.h
"""
SRC = 0
DST = 1
EDGE = 2
CODE2STR = {
0: "u",
1: "v",
2: "e",
}
class BuiltinFunction(object):
"""Base builtin function class."""
@property
def name(self):
"""Return the name of this builtin function."""
raise NotImplementedError