there are a few different test cases here that can be parameterized via tvm.testing.parameter. at least these, please check the rest of the file too: test_forward_add_layer_params test_forward_multiply_layer_params test_forward_concat_layer_params