The graph content generated by relay.build_module.build() is as following:
I don’t know which field represents input name.
In addition, I added a few lines of code, which have a background color, to print the input layer information,
The printing contents show that the input name is input_1 with the shape (?, 32, 32,3)
input_1 (<tf.Tensor 'input_1:0' shape=(?, 32, 32, 3) dtype=float32>,)