MODELFILE_NAME="rsqrt_4d.tflite"