tflite quantization aware training