Skip to content

Commit

Permalink
update input data for test of dense
Browse files Browse the repository at this point in the history
  • Loading branch information
Valery Chernov committed Jan 11, 2023
1 parent 915fad7 commit cccd755
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/python/unittest/test_meta_schedule_16x4_integration.py
Original file line number Diff line number Diff line change
Expand Up @@ -112,8 +112,8 @@ def _relay_dense(m, n, k):
relay_mod = tvm.IRModule.from_expr(out)
data = np.random.uniform(1, 10, size=(m, k)).astype("uint8")
params = {
"weight": np.random.uniform(1, 10, size=(n, k)).astype("int8"),
"bias": np.random.uniform(1, 10, size=(n,)).astype("int32"),
"weight": np.random.uniform(-10, 10, size=(n, k)).astype("int8"),
"bias": np.random.uniform(-10, 10, size=(n,)).astype("int32"),
}

def f_check(lib, dev):
Expand Down

0 comments on commit cccd755

Please sign in to comment.