Insert a task to the low priority task queue when the criticality is …

…one of the fixed list of low priority criticalities and support padding the high priority batch with the unbatched tasks given via the ProcessBatchCallBack.

PiperOrigin-RevId: 615925049

tensorflow/core/kernels/BUILD

@@ -1645,34 +1645,19 @@ cc_library(
 
 tf_cc_test(
     name = "batch_kernels_test",
-    size = "medium",
+    size = "small",
     srcs = ["batch_kernels_test.cc"],
     features = ["-layering_check"],
     deps = [
         ":batch_kernel_test_util",
         ":batch_kernels",
-        ":cwise_op",
         ":function_ops",
         ":shape_ops",
-        "//tensorflow/core:core_cpu_lib",
-        "//tensorflow/core:framework",
-        "//tensorflow/core:lib",
-        "//tensorflow/core:math_ops_op_lib",
         "//tensorflow/core:test",
         "//tensorflow/core:test_main",
         "//tensorflow/core:testlib",
-        "//tensorflow/core/framework:types_proto_cc",
         "//tensorflow/core/kernels/batching_util:warmup",
-        "//tensorflow/core/platform:status",
-        "//tensorflow/core/protobuf:for_core_protos_cc",
-        "//tensorflow/core/public:version",
-        "@com_google_absl//absl/status",
-        "@com_google_absl//absl/strings",
-        "@com_google_googletest//:gtest_main",
         "@local_tsl//tsl/platform:blocking_counter",
-        "@local_tsl//tsl/platform:errors",
-        "@local_tsl//tsl/platform:refcount",
-        "@local_tsl//tsl/platform:status",
     ],
 )