Can you share the arguments to nnvm.compiler.build(
?
Hope you called below API (or alternatively use add_shapes=True
while exporting protobuf from tensorflow).
# Add shapes to the graph.
with tf.Session() as sess:
graph_def = nnvm.testing.tf.AddShapesToGraphDef(sess, 'softmax')