[xla:runtime] Optimize arguments of single type
[xla:runtime] Optimize arguments of single type

Skip virtual dispatch if type is known at compile time.

name                       old cpu/op  new cpu/op  delta
BM_CreateMemrefDesc_1d/1   4.72ns ± 3%  4.61ns ±15%   -2.36%  (p=0.007 n=20+20)
BM_CreateMemrefDesc_1d/4   27.1ns ± 3%  25.8ns ±16%   -4.47%  (p=0.008 n=18+20)
BM_CreateMemrefDesc_1d/8   45.1ns ±10%  40.4ns ±16%  -10.46%  (p=0.000 n=18+20)
BM_CreateMemrefDesc_1d/12  66.2ns ± 4%  51.9ns ± 3%  -21.65%  (p=0.000 n=17+18)
BM_CreateMemrefDesc_1d/16  90.6ns ±17%  67.2ns ± 9%  -25.81%  (p=0.000 n=20+17)

PiperOrigin-RevId: 486674683

yongtang/tensorflow,tensorflow/tensorflow-pywrap_saved_model,tensorflow/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow,yongtang/tensorflow,paolodedios/tensorflow,yongtang/tensorflow,paolodedios/tensorflow,tensorflow/tensorflow-pywrap_saved_model,yongtang/tensorflow,paolodedios/tensorflow,Intel-tensorflow/tensorflow,Intel-tensorflow/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,yongtang/tensorflow,yongtang/tensorflow,tensorflow/tensorflow,yongtang/tensorflow,tensorflow/tensorflow-pywrap_saved_model,Intel-tensorflow/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,paolodedios/tensorflow,paolodedios/tensorflow,tensorflow/tensorflow-pywrap_saved_model,paolodedios/tensorflow,tensorflow/tensorflow-pywrap_saved_model,yongtang/tensorflow,tensorflow/tensorflow,yongtang/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,tensorflow/tensorflow-pywrap_saved_model,yongtang/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,paolodedios/tensorflow,tensorflow/tensorflow,Intel-tensorflow/tensorflow,Intel-tensorflow/tensorflow,paolodedios/tensorflow,tensorflow/tensorflow,paolodedios/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow-pywrap_saved_model,paolodedios/tensorflow,tensorflow/tensorflow,Intel-tensorflow/tensorflow,tensorflow/tensorflow,tensorflow/tensorflow,yongtang/tensorflow,paolodedios/tensorflow