From 4edc56419485660a673cd045dc521b847260283f Mon Sep 17 00:00:00 2001 From: Ryan Kuester Date: Thu, 10 Oct 2024 12:24:53 -0500 Subject: [PATCH] build(bazel): fix compression-related dependencies of micro_allocator --- tensorflow/lite/micro/BUILD | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/lite/micro/BUILD b/tensorflow/lite/micro/BUILD index f6922c24aa2..8cc9a19d304 100644 --- a/tensorflow/lite/micro/BUILD +++ b/tensorflow/lite/micro/BUILD @@ -165,6 +165,7 @@ tflm_cc_library( tflm_cc_library( name = "micro_allocator", srcs = [ + "compression.h", "micro_allocation_info.cc", "micro_allocator.cc", ], @@ -185,6 +186,7 @@ tflm_cc_library( "//tensorflow/lite/micro/arena_allocator:non_persistent_arena_buffer_allocator", "//tensorflow/lite/micro/arena_allocator:persistent_arena_buffer_allocator", "//tensorflow/lite/micro/arena_allocator:simple_memory_allocator", + "//tensorflow/lite/micro/compression:metadata_saved", "//tensorflow/lite/micro/memory_planner:greedy_memory_planner", "//tensorflow/lite/micro/memory_planner:linear_memory_planner", "//tensorflow/lite/micro/memory_planner:micro_memory_planner",