Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save vanbasten23/f41b43837be68742a2ca265bc62c49c5 to your computer and use it in GitHub Desktop.
Save vanbasten23/f41b43837be68742a2ca265bc62c49c5 to your computer and use it in GitHub Desktop.
This file has been truncated, but you can view the full file.
WARNING:root:libtpu.so and TPU device found. Setting PJRT_DEVICE=TPU.
INFO 07-16 22:24:48 [__init__.py:253] Automatically detected platform tpu.
INFO 07-16 22:24:48 [tpu.py:196] tpu_commons not found, using vLLM's TpuPlatform
/home/xiowei/miniconda3/envs/vllm312/lib/python3.12/site-packages/pytest_asyncio/plugin.py:208: PytestDeprecationWarning: The configuration option "asyncio_default_fixture_loop_scope" is unset.
The event loop scope for asynchronous fixtures will default to the fixture caching scope. Future versions of pytest-asyncio will default the loop scope for asynchronous fixtures to function scope. Set the default fixture loop scope explicitly in order to avoid unexpected behavior in the future. Valid fixture loop scopes are: "function", "class", "module", "package", "session"
warnings.warn(PytestDeprecationWarning(_DEFAULT_FIXTURE_LOOP_SCOPE_UNSET))
============================= test session starts ==============================
platform linux -- Python 3.12.11, pytest-8.3.3, pluggy-1.5.0 -- /home/xiowei/miniconda3/envs/vllm312/bin/python3.12
cachedir: .pytest_cache
hypothesis profile 'default' -> database=DirectoryBasedExampleDatabase(PosixPath('/home/xiowei/vllm/.hypothesis/examples'))
rootdir: /home/xiowei/vllm
configfile: pyproject.toml
plugins: shard-0.1.2, buildkite-test-collector-0.1.9, asyncio-0.24.0, mock-3.14.0, schemathesis-3.39.15, rerunfailures-14.0, subtests-0.14.1, forked-1.6.0, anyio-4.6.2.post1, timeout-2.3.1, hypothesis-6.131.0
asyncio: mode=Mode.STRICT, default_loop_scope=None
collecting ... collected 3 items / 2 deselected / 1 selected
Running 1 items in this shard: tests/tpu/lora/test_lora.py::test_single_lora
tests/tpu/lora/test_lora.py::test_single_lora xw32 setting up vLLM in test
INFO 07-16 22:24:56 [config.py:1561] Using max model len 256
INFO 07-16 22:24:56 [config.py:2357] Chunked prefill is enabled with max_num_batched_tokens=8192.
WARNING 07-16 22:24:56 [config.py:2404] max_num_batched_tokens (8192) exceeds max_num_seqs * max_model_len (2048). This may lead to unexpected behavior.
WARNING 07-16 22:24:56 [config.py:2404] max_num_batched_tokens (8192) exceeds max_num_seqs * max_model_len (2048). This may lead to unexpected behavior.
INFO 07-16 22:24:56 [tpu.py:110] [TPU] Forcing DYNAMO_ONCE compilation level
INFO 07-16 22:24:56 [core.py:532] Waiting for init message from front-end.
INFO 07-16 22:24:56 [core.py:69] Initializing a V1 LLM engine (v0.8.5.dev1940+ga937a3e99) with config: model='Qwen/Qwen2.5-3B-Instruct', speculative_config=None, tokenizer='Qwen/Qwen2.5-3B-Instruct', skip_tokenizer_init=False, tokenizer_mode=auto, revision=None, override_neuron_config={}, tokenizer_revision=None, trust_remote_code=False, dtype=torch.bfloat16, max_seq_len=256, download_dir=None, load_format=LoadFormat.AUTO, tensor_parallel_size=1, pipeline_parallel_size=1, disable_custom_all_reduce=True, quantization=None, enforce_eager=False, kv_cache_dtype=auto, device_config=None, decoding_config=DecodingConfig(backend='auto', disable_fallback=False, disable_any_whitespace=False, disable_additional_properties=False, reasoning_backend=''), observability_config=ObservabilityConfig(show_hidden_metrics_for_version=None, otlp_traces_endpoint=None, collect_detailed_traces=None), seed=0, served_model_name=Qwen/Qwen2.5-3B-Instruct, num_scheduler_steps=1, multi_step_stream_outputs=True, enable_prefix_caching=True, chunked_prefill_enabled=True, use_async_output_proc=False, pooler_config=None, compilation_config={"level":2,"debug_dump_path":"","cache_dir":"","backend":"openxla","custom_ops":[],"splitting_ops":["vllm.unified_attention","vllm.unified_attention_with_output"],"use_inductor":true,"compile_sizes":[],"inductor_compile_config":{"enable_auto_functionalized_v2":false},"inductor_passes":{},"use_cudagraph":true,"cudagraph_num_of_warmups":1,"cudagraph_capture_sizes":[16,8,4,2,1],"cudagraph_copy_inputs":false,"full_cuda_graph":false,"max_capture_size":16,"local_cache_dir":null}
INFO 07-16 22:24:56 [importing.py:43] Triton is installed but 0 active driver(s) found (expected 1). Disabling Triton to prevent runtime errors.
INFO 07-16 22:24:56 [importing.py:63] Triton not installed or not compatible; certain GPU-related functions will not be available.
INFO 07-16 22:24:56 [tpu_worker.py:320] tpu_commons not found, using vLLM's TPUWorker.
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
[Gloo] Rank 0 is connected to 0 peer ranks. Expected number of connected peer ranks is : 0
INFO 07-16 22:24:56 [parallel_state.py:1090] rank 0 in world size 1 is assigned as DP rank 0, PP rank 0, TP rank 0, EP rank 0
xw32 TPUModelRunner.__init__
WARNING 07-16 22:25:23 [tpu.py:163] Pin memory is not supported on TPU.
INFO 07-16 22:25:23 [tpu_model_runner.py:1818] Using exponential token paddings:
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 16
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 32
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 64
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 128
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 256
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 512
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 1024
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 2048
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 4096
INFO 07-16 22:25:23 [tpu_model_runner.py:1820] 8192
INFO 07-16 22:25:23 [tpu_model_runner.py:1784] Preparing request paddings:
INFO 07-16 22:25:23 [tpu_model_runner.py:1791] 8
INFO 07-16 22:25:23 [tpu_model_runner.py:1168] Loading model from scratch...
INFO 07-16 22:25:23 [tpu.py:51] Cannot use None backend on TPU.
INFO 07-16 22:25:23 [tpu.py:55] Using Pallas V1 backend.
INFO 07-16 22:25:24 [weight_utils.py:296] Using model weights format ['*.safetensors']
Loading safetensors checkpoint shards: 0% Completed | 0/2 [00:00<?, ?it/s]
2025-07-16 22:25:24.381497: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:25.158630: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:25.409684: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:25.649584: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:25.670155: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:25.811523: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:25.830282: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:26.059865: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:26.080268: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
Loading safetensors checkpoint shards: 50% Completed | 1/2 [00:02<00:02, 2.98s/it]
Loading safetensors checkpoint shards: 100% Completed | 2/2 [00:03<00:00, 1.58s/it]
Loading safetensors checkpoint shards: 100% Completed | 2/2 [00:03<00:00, 1.79s/it]
INFO 07-16 22:25:27 [default_loader.py:272] Loading weights took 3.66 seconds
xw32 LoRAModelRunnerMixin.load_lora_model
xw32 PunicaWrappterTPU.__init__
INFO 07-16 22:25:28 [punica_selector.py:19] Using PunicaWrapperTPU.
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__2025-07-16 22:25:28.201515: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.485011: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.503860: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.525238: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.537793: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.551888: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.566364: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.585228: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:28.598212: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedQKVParallelLinearWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedQKVParallelLinearWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 MergedColumnParallelLinearWithLoRA.__init__
xw32 ColumnParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 MergedColumnParallelLinearWithLoRA.create_lora_weights
xw32 RowParallelLinearWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.__init__
xw32 BaseLinearLayerWithLoRA.create_lora_weights
xw32 LoRAModelRunnerMixin.maybe_setup_dummy_loras
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora2025-07-16 22:25:28.841109: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:29 [kv_cache_utils.py:732] GPU KV cache size: 646,368 tokens
INFO 07-16 22:25:29 [kv_cache_utils.py:736] Maximum concurrency for 256 tokens per request: 2524.88x
xw32 LoRAModelRunnerMixin.maybe_setup_dummy_loras
INFO 07-16 22:25:30 [tpu_model_runner.py:1342] Compiling the model with different input shapes.
INFO 07-16 22:25:30 [tpu_model_runner.py:1345] -- num_tokens: 16
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora2025-07-16 22:25:30.757781: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:31 [tpu_model_runner.py:1345] -- num_tokens: 32
2025-07-16 22:25:31.333714: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:32.347519: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:32 [tpu_model_runner.py:1345] -- num_tokens: 64
2025-07-16 22:25:32.799760: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:33.782212: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:34 [tpu_model_runner.py:1345] -- num_tokens: 128
2025-07-16 22:25:34.215521: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:35.238949: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:35 [tpu_model_runner.py:1345] -- num_tokens: 256
2025-07-16 22:25:35.692425: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:36.871454: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:37 [tpu_model_runner.py:1345] -- num_tokens: 512
2025-07-16 22:25:37.316242: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:38.881361: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:39 [tpu_model_runner.py:1345] -- num_tokens: 1024
2025-07-16 22:25:39.322536: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:41.272474: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:42 [tpu_model_runner.py:1345] -- num_tokens: 2048
2025-07-16 22:25:42.138292: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:44.333816: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:44 [tpu_model_runner.py:1345] -- num_tokens: 4096
2025-07-16 22:25:44.801554: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:48.462107: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:48 [tpu_model_runner.py:1345] -- num_tokens: 8192
2025-07-16 22:25:48.918056: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 TPUModelRunner._dummy_run
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
INFO 07-16 22:25:51 [tpu_model_runner.py:1353] Compilation finished in 21.42 [secs].
INFO 07-16 22:25:51 [tpu_model_runner.py:1359] Compiling select_hidden_states with different input shapes.
2025-07-16 22:25:54.167716: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.211402: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 16, num_seqs: 8
2025-07-16 22:25:58.245423: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.260377: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 32, num_seqs: 8
2025-07-16 22:25:58.291645: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.307273: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 64, num_seqs: 8
2025-07-16 22:25:58.342485: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.359847: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 128, num_seqs: 8
2025-07-16 22:25:58.401849: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.422322: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 256, num_seqs: 8
2025-07-16 22:25:58.481480: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.509811: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 512, num_seqs: 8
2025-07-16 22:25:58.601916: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.679739: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:58 [tpu_model_runner.py:1374] -- num_tokens: 1024, num_seqs: 8
2025-07-16 22:25:58.836712: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:58.855703: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:59 [tpu_model_runner.py:1374] -- num_tokens: 2048, num_seqs: 8
2025-07-16 22:25:59.147202: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:59.166374: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:59 [tpu_model_runner.py:1374] -- num_tokens: 4096, num_seqs: 8
2025-07-16 22:25:59.189621: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:59.207179: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:59 [tpu_model_runner.py:1374] -- num_tokens: 8192, num_seqs: 8
INFO 07-16 22:25:59 [tpu_model_runner.py:1382] Compilation finished in 7.29 [secs].
INFO 07-16 22:25:59 [tpu_model_runner.py:1386] Compiling compute_logits with different input shapes.
2025-07-16 22:25:59.333067: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:59.351972: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:25:59 [tpu_model_runner.py:1395] -- num_seqs: 8
INFO 07-16 22:25:59 [tpu_model_runner.py:1398] Compilation finished in 0.34 [secs].
INFO 07-16 22:25:59 [tpu_model_runner.py:1402] Compiling structured_decoding with different input shapes.
2025-07-16 22:25:59.763164: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:25:59.802135: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
INFO 07-16 22:26:00 [tpu_model_runner.py:1419] -- num_seqs: 8
INFO 07-16 22:26:00 [tpu_model_runner.py:1422] Compilation finished in 0.69 [secs].
INFO 07-16 22:26:00 [tpu_model_runner.py:1426] Compiling sample_from_logits with different input shapes.
2025-07-16 22:26:00.258498: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:26:00.426033: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:26:07.156660: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
INFO 07-16 22:26:07 [tpu_model_runner.py:1450] -- num_seqs: 8
INFO 07-16 22:26:07 [tpu_model_runner.py:1453] Compilation finished in 7.06 [secs].
INFO 07-16 22:26:07 [tpu_model_runner.py:1457] Compiling gather_logprobs with different input shapes.
2025-07-16 22:26:07.376574: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 LoRAModelRunnerMixin.maybe_select_dummy_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
INFO 07-16 22:26:08 [tpu_model_runner.py:1468] -- num_seqs: 8
INFO 07-16 22:26:08 [tpu_model_runner.py:1471] Compilation finished in 0.70 [secs].
INFO 07-16 22:26:08 [core.py:178] init engine (profile, create kv cache, warmup model) took 39.58 seconds
INFO 07-16 22:26:08 [tpu.py:110] [TPU] Forcing DYNAMO_ONCE compilation level
xw32 calling llm.generate in test
Adding requests: 0%| | 0/1 [00:00<?, ?it/s]WARNING 07-16 22:26:08 [tokenizer.py:299] No tokenizer found in Username6568/Qwen2.5-3B-Instruct-1_plus_1_equals_1_adapter, using base model tokenizer instead. (Exception: Unrecognized model in Username6568/Qwen2.5-3B-Instruct-1_plus_1_equals_1_adapter. Should have a `model_type` key in its config.json, or contain one of the following strings in its name: albert, align, altclip, arcee, aria, aria_text, audio-spectrogram-transformer, autoformer, aya_vision, bamba, bark, bart, beit, bert, bert-generation, big_bird, bigbird_pegasus, biogpt, bit, bitnet, blenderbot, blenderbot-small, blip, blip-2, blip_2_qformer, bloom, bridgetower, bros, camembert, canine, chameleon, chinese_clip, chinese_clip_vision_model, clap, clip, clip_text_model, clip_vision_model, clipseg, clvp, code_llama, codegen, cohere, cohere2, colpali, colqwen2, conditional_detr, convbert, convnext, convnextv2, cpmant, csm, ctrl, cvt, d_fine, dab-detr, dac, data2vec-audio, data2vec-text, data2vec-vision, dbrx, deberta, deberta-v2, decision_transformer, deepseek_v3, deformable_detr, deit, depth_anything, depth_pro, deta, detr, dia, diffllama, dinat, dinov2, dinov2_with_registers, distilbert, donut-swin, dots1, dpr, dpt, efficientformer, efficientnet, electra, emu3, encodec, encoder-decoder, ernie, ernie_m, esm, falcon, falcon_h1, falcon_mamba, fastspeech2_conformer, flaubert, flava, fnet, focalnet, fsmt, funnel, fuyu, gemma, gemma2, gemma3, gemma3_text, gemma3n, gemma3n_audio, gemma3n_text, gemma3n_vision, git, glm, glm4, glm4v, glm4v_text, glpn, got_ocr2, gpt-sw3, gpt2, gpt_bigcode, gpt_neo, gpt_neox, gpt_neox_japanese, gptj, gptsan-japanese, granite, granite_speech, granitemoe, granitemoehybrid, granitemoeshared, granitevision, graphormer, grounding-dino, groupvit, helium, hgnet_v2, hiera, hubert, ibert, idefics, idefics2, idefics3, idefics3_vision, ijepa, imagegpt, informer, instructblip, instructblipvideo, internvl, internvl_vision, jamba, janus, jetmoe, jukebox, kosmos-2, kyutai_speech_to_text, layoutlm, layoutlmv2, layoutlmv3, led, levit, lightglue, lilt, llama, llama4, llama4_text, llava, llava_next, llava_next_video, llava_onevision, longformer, longt5, luke, lxmert, m2m_100, mamba, mamba2, marian, markuplm, mask2former, maskformer, maskformer-swin, mbart, mctct, mega, megatron-bert, mgp-str, mimi, minimax, mistral, mistral3, mixtral, mlcd, mllama, mobilebert, mobilenet_v1, mobilenet_v2, mobilevit, mobilevitv2, modernbert, moonshine, moshi, mpnet, mpt, mra, mt5, musicgen, musicgen_melody, mvp, nat, nemotron, nezha, nllb-moe, nougat, nystromformer, olmo, olmo2, olmoe, omdet-turbo, oneformer, open-llama, openai-gpt, opt, owlv2, owlvit, paligemma, patchtsmixer, patchtst, pegasus, pegasus_x, perceiver, persimmon, phi, phi3, phi4_multimodal, phimoe, pix2struct, pixtral, plbart, poolformer, pop2piano, prompt_depth_anything, prophetnet, pvt, pvt_v2, qdqbert, qwen2, qwen2_5_omni, qwen2_5_vl, qwen2_5_vl_text, qwen2_audio, qwen2_audio_encoder, qwen2_moe, qwen2_vl, qwen2_vl_text, qwen3, qwen3_moe, rag, realm, recurrent_gemma, reformer, regnet, rembert, resnet, retribert, roberta, roberta-prelayernorm, roc_bert, roformer, rt_detr, rt_detr_resnet, rt_detr_v2, rwkv, sam, sam_hq, sam_hq_vision_model, sam_vision_model, seamless_m4t, seamless_m4t_v2, segformer, seggpt, sew, sew-d, shieldgemma2, siglip, siglip2, siglip_vision_model, smollm3, smolvlm, smolvlm_vision, speech-encoder-decoder, speech_to_text, speech_to_text_2, speecht5, splinter, squeezebert, stablelm, starcoder2, superglue, superpoint, swiftformer, swin, swin2sr, swinv2, switch_transformers, t5, t5gemma, table-transformer, tapas, textnet, time_series_transformer, timesfm, timesformer, timm_backbone, timm_wrapper, trajectory_transformer, transfo-xl, trocr, tvlt, tvp, udop, umt5, unispeech, unispeech-sat, univnet, upernet, van, video_llava, videomae, vilt, vipllava, vision-encoder-decoder, vision-text-dual-encoder, visual_bert, vit, vit_hybrid, vit_mae, vit_msn, vitdet, vitmatte, vitpose, vitpose_backbone, vits, vivit, vjepa2, wav2vec2, wav2vec2-bert, wav2vec2-conformer, wavlm, whisper, xclip, xglm, xlm, xlm-prophetnet, xlm-roberta, xlm-roberta-xl, xlnet, xmod, yolos, yoso, zamba, zamba2, zoedepth, siglip_visual_tokenizer, aimv2_visual_tokenizer)
Adding requests: 100%|██████████| 1/1 [00:00<00:00, 8.78it/s]
Adding requests: 100%|██████████| 1/1 [00:00<00:00, 8.76it/s]
Processed prompts: 0%| | 0/1 [00:00<?, ?it/s, est. speed input: 0.00 toks/s, output: 0.00 toks/s]xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
Fetching 5 files: 0%| | 0/5 [00:00<?, ?it/s]
Fetching 5 files: 100%|██████████| 5/5 [00:00<00:00, 10849.21it/s]
2025-07-16 22:26:08.566503: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 MergedColumnParallelLinearWithLoRA.set_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 BaseLinearLayerWithLoRA.reset_lora
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice2025-07-16 22:26:08.735432: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
2025-07-16 22:26:09.839545: W torch_xla/csrc/runtime/pjrt_computation_client.cpp:683] Failed to deserialize executable: UNIMPLEMENTED: Deserializing serialized executable not supported.
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 PunicaWrappterTPU.expand_slice
xw32 RowParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappterTPU.add_lora_linear
xw32 PunicaWrappterTPU.add_shrink
xw32 PunicaWrappterTPU.shrink
xw32 PunicaWrappterTPU.add_expand
xw32 PunicaWrappterTPU.expand_slice
xw32 TPUModelRunner.execute_model
xw32 TPUModelRunner._prepare_inputs
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 LoRAModelRunnerMixin.set_active_loras
xw32 PunicaWrappterTPU._update_base_metadata
xw32 PunicaWrappterTPU._pad_prompt_mapping
xw32 PunicaWrappterTPU._update_prefill_metadata
xw32 ColumnParallelLinearWithLoRA.forward
xw32 BaseLinearLayerWithLoRA.apply
xw32 PunicaWrappter
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment