diff --git a/runtime/core/targets.bzl b/runtime/core/targets.bzl index 61351b083bf..c3535688f63 100644 --- a/runtime/core/targets.bzl +++ b/runtime/core/targets.bzl @@ -41,7 +41,6 @@ def define_common_targets(): "defines.h", "error.h", "freeable_buffer.h", - "named_data_map.h", "result.h", "span.h", ], @@ -133,6 +132,20 @@ def define_common_targets(): ], ) + runtime.cxx_library( + name = "named_data_map", + exported_headers = [ + "named_data_map.h", + ], + visibility = [ + "//executorch/...", + "@EXECUTORCH_CLIENTS", + ], + exported_deps = [ + ":tensor_layout", + ], + ) + runtime.cxx_library( name = "tensor_layout", srcs = ["tensor_layout.cpp"], diff --git a/runtime/core/tensor_layout.cpp b/runtime/core/tensor_layout.cpp index 07106e6c7e4..748a43fc5d6 100644 --- a/runtime/core/tensor_layout.cpp +++ b/runtime/core/tensor_layout.cpp @@ -30,7 +30,7 @@ Result calculate_nbytes( } } // namespace -Result TensorLayout::create( +Result TensorLayout::create( Span sizes, Span dim_order, executorch::aten::ScalarType scalar_type) { diff --git a/runtime/core/tensor_layout.h b/runtime/core/tensor_layout.h index dcd0a2af839..6ad04f19fbd 100644 --- a/runtime/core/tensor_layout.h +++ b/runtime/core/tensor_layout.h @@ -33,7 +33,7 @@ class ET_EXPERIMENTAL TensorLayout final { * @param[in] scalar_type The scalar type of the tensor. * @return A Result containing the TensorLayout on success, or an error. */ - static executorch::runtime::Result create( + static executorch::runtime::Result create( Span sizes, Span dim_order, executorch::aten::ScalarType scalar_type);