And restore -layering_check to a few places where it got removed with https://github.com/tensorflow/tflite-micro/pull/435/

BUG=http://b/177257332