[EdgeTPU] Add ability to dynamically change partition/edge TPU config

[BrendanCReidy]

This would allow for training using a normal model, and then the model can be partitioned, use conv dot products, use I-GELU, etc after training. This would allow for faster training and fix some bugs associated with training a partitioned model