def tvmgen_default_run_model(input: T.handle, output: T.handle) -> None:
# function attr dict
T.func_attr({
"global_symbol": "tvmgen_default_run_model",
"runner_function": True
})
# body
T.attr("default", "device_id", 0)
T.attr("default", "device_type", 1)
sid_9 = T.allocate([301056], "int8", "global")
sid_8 = T.allocate([802816], "int8", "global")
T.evaluate(
T.call_extern("tvmgen_default_fused_cast_subtract",
input,
T.lookup_param("p0", dtype="handle"),
sid_9,
dtype="int32"))
T.evaluate(
T.call_extern(
"tvmgen_default_fused_nn_conv2d_add_fixed_point_multiply_clip_cast",
sid_9,
T.lookup_param("p1", dtype="handle"),
T.lookup_param("p2", dtype="handle"),
sid_8,
dtype="int32"))
T.evaluate(
T.call_extern("tvmgen_default_fused_nn_max_pool2d_cast",
sid_8,
output,
dtype="int32"))
def tvmgen_default_run_model(input: T.handle, output: T.handle) -> None:
# function attr dict
T.func_attr({
"global_symbol": "tvmgen_default_run_model",
"runner_function": True
})
# body
T.attr("default", "device_id", 0)
T.attr("default", "device_type", 1)
sid_2 = T.allocate([720000], "int8", "global")
sid_6 = T.allocate([5760000], "int8", "global")
sid_7 = T.allocate([720000], "int8", "global")
sid_8 = T.allocate([720000], "int8", "global")
T.evaluate(
T.call_extern(
"tvmgen_default_fused_cast_subtract_fixed_point_multiply_add_clip_cast_cast",
input,
T.lookup_param("p0", dtype="handle"),
sid_2.data,
dtype="int32"))
T.evaluate(
T.call_extern(
"tvmgen_default_fused_nn_conv2d_add_fixed_point_multiply_clip_cast_cast",
sid_2.data,
T.lookup_param("p3", dtype="handle"),
T.lookup_param("p4", dtype="handle"),
sid_8.data,
dtype="int32"))
T.evaluate(
T.call_extern(
"tvmgen_default_fused_nn_conv2d_add_fixed_point_multiply_clip_cast_cast_1",
sid_8.data,
T.lookup_param("p5", dtype="handle"),
T.lookup_param("p6", dtype="handle"),
sid_7.data,
dtype="int32"))
T.evaluate(
T.call_extern(
"tvmgen_default_fused_nn_conv2d_add_fixed_point_multiply_add_clip_cast_cast_subtract_fixed_point_15934180698220515269_",
sid_7.data,
T.lookup_param("p7", dtype="handle"),
T.lookup_param("p8", dtype="handle"),
sid_6.data,
dtype="int32"))
T.evaluate(
T.call_extern(
"tvmgen_default_fused_nn_conv2d_add_fixed_point_multiply_add_clip_cast_cast_subtract_fixed_point_4200876283395191415_",
sid_2.data,
T.lookup_param("p1", dtype="handle"),
T.lookup_param("p2", dtype="handle"),
sid_6.data,
output,
dtype="int32"))
def __tvm_main__(input1: T.handle, input2: T.handle, output1: T.handle, output2: T.handle) -> None:
# function attr dict
T.func_attr({"global_symbol": "__tvm_main__", "runner_function": True})
input1_buffer_var = T.match_buffer(input1, [150528], dtype="uint8", elem_offset=0, align=128, offset_factor=1)
input2_buffer_var = T.match_buffer(input2, [150528], dtype="uint8", elem_offset=0, align=128, offset_factor=1)
output1_buffer_var = T.match_buffer(output1, [452], dtype="int16", elem_offset=0, align=128, offset_factor=1)
output2_buffer_var = T.match_buffer(output2, [452], dtype="int16", elem_offset=0, align=128, offset_factor=1)
# body
T.evaluate(T.call_extern("tvmgen_default_fused_cast_subtract", input1_buffer_var.data, T.lookup_param("p0", dtype="handle"), output1_buffer_var.data, dtype="int32"))
T.evaluate(T.call_extern("tvmgen_default_fused_cast_subtract", input2_buffer_var.data, T.lookup_param("p1", dtype="handle"), output2_buffer_var.data, dtype="int32"))
