This thread has been locked.

If you have a related question, please click the "Ask a related question" button in the top right corner. The newly created question will be automatically linked to this question.

TDA2: Problem in TDA2: Example to train CNN network on tensorflow and run using TIDL

Part Number: TDA2

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 ?