[tfrt:jitrt] Optimize memref arguments packing
[tfrt:jitrt] Optimize memref arguments packing

name                                      old cpu/op  new cpu/op  delta
BM_InitializeCallFrameUniqueShapes_false  25.1ns ± 0%  15.9ns ± 0%  -36.80%
BM_InitializeCallFrameSameShapes_false    25.1ns ± 0%  15.9ns ± 0%  -36.88%
BM_InitializeCallFrameKnownShapes_false   25.1ns ± 0%  15.8ns ± 0%  -36.88%

PiperOrigin-RevId: 445444955

tensorflow/runtime,tensorflow/runtime,tensorflow/runtime,tensorflow/runtime