Depend on granular targets rather than `//third_party/tensorflow/core:framework_headers_lib`

PiperOrigin-RevId: 884615862
diff --git a/c/enc/matching_tag_mask.h b/c/enc/matching_tag_mask.h
index 8a1be1d..c16400e 100644
--- a/c/enc/matching_tag_mask.h
+++ b/c/enc/matching_tag_mask.h
@@ -28,7 +28,7 @@
   }
 #else
   const int chunk_size = sizeof(size_t);
-  const size_t shift_amount = ((chunk_size * 8) - chunk_size);
+  const ssize_t shift_amount = ((chunk_size * 8) - chunk_size);
   const size_t xFF = ~((size_t)0);
   const size_t x01 = xFF / 0xFF;
   const size_t x80 = x01 << 7;