def METHOD_NAME():
    aot_executor = tvm.runtime.executor.aot_executor.AotModule(session.create_aot_executor())
    assert aot_executor.get_input_index("a") == 0
    assert aot_executor.get_input_index("b") == 1
    assert aot_executor.get_num_inputs() == 2
    assert aot_executor.get_num_outputs() == 1
    A_np = np.array([[2, 3]], dtype="uint8")
    B_np = np.array([[4, 7]], dtype="uint8")
    A_data = aot_executor.get_input("a").copyfrom(A_np)
    B_data = aot_executor.get_input("b").copyfrom(B_np)
    aot_executor.run()
    out = aot_executor.get_output(0)
    assert (out.numpy() == np.array([6, 10])).all()
    B_np_new = np.array([[5, 8]])
    aot_executor.set_input("b", B_np_new)
    assert (B_data.numpy() == B_np_new).all()