dmlc--dgl
Hongzhi (Steve), Chen
98325b1097
Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
16 行
252 B
Python
16 行
252 B
Python
import time
|
|
import unittest
|
|
from timeit import default_timer
|
|
|
|
import backend as F
|
|
import numpy as np
|
|
import pytest
|
|
from test_utils import get_cases, parametrize_idtype
|
|
|
|
import dgl
|
|
import dgl.function as fn
|
|
|
|
iters = 5
|
|
n_edge_scale = 1
|
|
num_rel_scale = 1
|