Add Tensor Map kernels to Flex

TensorMap kernels are typical examples of variants. Those kernels can be used
for verifying the implementations of supporting resources and variants in
TFLite.

PiperOrigin-RevId: 351931000
Change-Id: I0d2348d0449994e3dfbbf5fa6d7360453b4aab52

tensorflow/core/kernels/BUILD

@@ -5914,6 +5914,7 @@ filegroup(
         "lookup_table_op.h",
         "lookup_util.h",
         "list_kernels.h",
+        "map_kernels.h",
         "maxpooling_op.h",
         "mfcc.h",
         "mfcc_dct.h",
@@ -5949,6 +5950,7 @@ filegroup(
         "string_to_hash_bucket_fast_op.h",
         "tensor_array.h",
         "tensor_list.h",
+        "tensor_map.h",
         "tile_functor.h",
         "tile_ops_impl.h",
         "topk_op.h",
@@ -6139,6 +6141,7 @@ filegroup(
         "lookup_table_op.cc",
         "lookup_util.cc",
         "lrn_op.cc",
+        "map_kernels.cc",
         "maxpooling_op.cc",
         "mfcc.cc",
         "mfcc_dct.cc",
@@ -6225,6 +6228,7 @@ filegroup(
         "tensor_array.cc",
         "tensor_array_ops.cc",
         "tensor_list.cc",
+        "tensor_map.cc",
         "tile_functor_cpu.h",
         "tile_functor_cpu_bfloat16.cc",
         "tile_functor_cpu_bool.cc",

tensorflow/lite/delegates/flex/allowlisted_flex_ops.cc

@@ -166,6 +166,7 @@ const std::set<std::string>& GetFlexAllowlist() {
           "EluGrad",
           "Empty",
           "EmptyTensorList",
+          "EmptyTensorMap",
           "EncodeBase64",
           "EncodeJpeg",
           "EncodeJpegVariableQuality",
@@ -621,6 +622,12 @@ const std::set<std::string>& GetFlexAllowlist() {
           "TensorListSetItem",
           "TensorListSplit",
           "TensorListStack",
+          "TensorMapErase",
+          "TensorMapHasKey",
+          "TensorMapInsert",
+          "TensorMapLookup",
+          "TensorMapSize",
+          "TensorMapStackKeys",
           "TensorScatterAdd",
           "TensorScatterMax",
           "TensorScatterMin",