Python make_relu示例

示例#1

文件： test_tvm_op.py 项目： joshpoll/assignment2-2018

def test_relu():
    shape = (2000, 2500)
    x = np.random.uniform(-1, 1, shape).astype(dtype)
    y = np.zeros(shape).astype(dtype)
    arr_x = tvm.nd.array(x, ctx=ctx)
    arr_y = tvm.nd.array(y, ctx=ctx)
    relu = tvm_op.make_relu(shape, tgt, tgt_host, "relu")
    relu(arr_x, arr_y)
    y = arr_y.asnumpy()
    np.testing.assert_allclose(np.maximum(x, 0).astype(dtype), y)

示例#2

文件： test_tvm_op.py 项目： LibAssignment/dlsys-course-assignment2-2018

def test_relu():
    shape = (2000, 2500)
    x = np.random.uniform(-1, 1, shape).astype(dtype)
    y = np.zeros(shape).astype(dtype)
    arr_x = tvm.nd.array(x, ctx=ctx)
    arr_y = tvm.nd.array(y, ctx=ctx)
    relu = tvm_op.make_relu(shape, tgt, tgt_host, "relu")
    relu(arr_x, arr_y)
    y = arr_y.asnumpy()
    np.testing.assert_allclose(np.maximum(x, 0).astype(dtype), y)