[Pallas] Support Dynamo (#6477)
Summary:
This pull request enables dynamo support for custom tpu calls, e.g. ones written in Pallas.
Test Plan:
PJRT_DEVICE=TPU XLA_DISABLE_FUNCTIONALIZATION=1 python test/test_operations.py -v -k test_tpu_custom_call_pallas_add_one_dynamo
