Function _make_dataset

tests/python/relax/test_training_trainer_numeric.py:51–53

()

def _make_dataset():
    N = 100
    return [[np.ones((1, 10)).astype(np.float32), np.array([[0, 0, 1, 0, 0]], np.float32)]] * N


@tvm.testing.parametrize_targets("llvm")