test_batch_norm_setup
test_conv_setup
The function to test conv api setup
test_dense_setup
The function to test dense api setup