I follow the every step in https://e2e.ti.com/support/processors/f/791/t/689876
But, I failed in the last step when I performed the command below
python -m tensorflow.python.tools.optimize_for_inference --input=keras_frozen.pb --output=keras_frozen_optimized.pb --input_names=conv2d_1_input --output_name=conv2d_5/BiasAdd
and get the error message below.
KeyError: "The following input nodes were not found: {'conv2d_1_input'}\n"
It seems that the setting of conv2d_1_input is wrong.
How could I set it ?