Litert Enable TF OP support (Flex delegate) pip package

It used to be fairly easy to build and cross compile the tflite runtime pip package with the flex delegate enabled.

CI_DOCKER_EXTRA_PARAMS="-e CUSTOM_BAZEL_FLAGS=--define=tflite_pip_with_flex=true" \
  tensorflow/tools/ci_build/ci_build.sh PI-PYTHON39 \
  tensorflow/lite/tools/pip_package/build_pip_package_with_bazel.sh aarch64

Which worked at the time but seems to of stopped at python3.9 as replacing with 311 cause an error

In the newer LiteRT documentation we have

make -C tensorflow/lite/tools/pip_package docker-build \
  TENSORFLOW_TARGET=aarch64 PYTHON_VERSION=3.11

Which works but I can not find any reference of doing “CUSTOM_BAZEL_FLAGS=–define=tflite_pip_with_flex=true” with the newer method?