Debug updates and updated tracing

• Issues with out-of-order module definitions resolved
• Tracing now runs a single forward pass iteration of the model
• CompressNNModel() now requires input shape to be specified (for use in tracing)