react-native-executorch 0.4.8 → 0.5.1-rc.0

package/third-party/ios/ExecutorchLib/frameworks/executorch.xcframework/ios-arm64-simulator/Headers/executorch/ExecuTorchValue.h

@@ -0,0 +1,219 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#import "ExecuTorchTensor.h"
+
+NS_ASSUME_NONNULL_BEGIN
+
+/**
+ * Enum to define the dynamic type of a Value.
+ * Values can be a subset, but must numerically match exactly those defined in
+ * runtime/core/tag.h
+ */
+typedef NS_ENUM(uint32_t, ExecuTorchValueTag) {
+  ExecuTorchValueTagNone,
+  ExecuTorchValueTagTensor,
+  ExecuTorchValueTagString,
+  ExecuTorchValueTagDouble,
+  ExecuTorchValueTagInteger,
+  ExecuTorchValueTagBoolean,
+  ExecuTorchValueTagBooleanList,
+  ExecuTorchValueTagDoubleList,
+  ExecuTorchValueTagIntegerList,
+  ExecuTorchValueTagTensorList,
+  ExecuTorchValueTagScalarList,
+  ExecuTorchValueTagOptionalTensorList,
+} NS_SWIFT_NAME(ValueTag);
+
+typedef NSNumber *
+    ExecuTorchScalarValue NS_SWIFT_BRIDGED_TYPEDEF NS_SWIFT_NAME(ScalarValue);
+typedef NSString *
+    ExecuTorchStringValue NS_SWIFT_BRIDGED_TYPEDEF NS_SWIFT_NAME(StringValue);
+typedef BOOL
+    ExecuTorchBooleanValue NS_SWIFT_BRIDGED_TYPEDEF NS_SWIFT_NAME(BoolValue);
+typedef NSInteger
+    ExecuTorchIntegerValue NS_SWIFT_BRIDGED_TYPEDEF NS_SWIFT_NAME(IntegerValue);
+typedef double
+    ExecuTorchDoubleValue NS_SWIFT_BRIDGED_TYPEDEF NS_SWIFT_NAME(DoubleValue);
+
+/**
+ * A dynamic value type used by ExecuTorch.
+ *
+ * ExecuTorchValue encapsulates a value that may be of various types such as
+ * a tensor or a scalar. The value’s type is indicated by its tag.
+ */
+NS_SWIFT_NAME(Value)
+__attribute__((deprecated("This API is experimental.")))
+@interface ExecuTorchValue : NSObject
+
+/**
+ * The tag that indicates the dynamic type of the value.
+ *
+ * @return An ExecuTorchValueTag value.
+ */
+@property(nonatomic, readonly) ExecuTorchValueTag tag;
+
+/**
+ * The tensor value if the tag is ExecuTorchValueTagTensor.
+ *
+ * @return A Tensor instance or nil.
+ */
+@property(nullable, nonatomic, readonly)
+    ExecuTorchTensor *tensorValue NS_SWIFT_NAME(tensor);
+
+/**
+ * The string value if the tag is ExecuTorchValueTagString.
+ *
+ * @return An NSString instance or nil.
+ */
+@property(nullable, nonatomic, readonly)
+    ExecuTorchStringValue stringValue NS_SWIFT_NAME(string);
+
+/**
+ * The scalar value if the tag is boolean, integer or double.
+ *
+ * @return A scalar value or nil.
+ */
+@property(nullable, nonatomic, readonly)
+    ExecuTorchScalarValue scalarValue NS_SWIFT_NAME(scalar);
+
+/**
+ * The boolean value if the tag is ExecuTorchValueTagBoolean.
+ *
+ * @return A BOOL representing the boolean value.
+ */
+@property(nonatomic, readonly)
+    ExecuTorchBooleanValue boolValue NS_SWIFT_NAME(boolean);
+
+/**
+ * The integer value if the tag is ExecuTorchValueTagInteger.
+ *
+ * @return An NSInteger representing the integer value.
+ */
+@property(nonatomic, readonly)
+    ExecuTorchIntegerValue intValue NS_SWIFT_NAME(integer);
+
+/**
+ * The double value if the tag is ExecuTorchValueTagDouble.
+ *
+ * @return A double representing the double value.
+ */
+@property(nonatomic, readonly)
+    ExecuTorchDoubleValue doubleValue NS_SWIFT_NAME(double);
+
+/**
+ * Returns YES if the value is of type None.
+ *
+ * @return A BOOL indicating whether the value is None.
+ */
+@property(nonatomic, readonly) BOOL isNone;
+
+/**
+ * Returns YES if the value is a Tensor.
+ *
+ * @return A BOOL indicating whether the value is a Tensor.
+ */
+@property(nonatomic, readonly) BOOL isTensor;
+
+/**
+ * Returns YES if the value is a string.
+ *
+ * @return A BOOL indicating whether the value is a string.
+ */
+@property(nonatomic, readonly) BOOL isString;
+
+/**
+ * Returns YES if the value is a scalar (boolean, integer or double).
+ *
+ * @return A BOOL indicating whether the value is a scalar.
+ */
+@property(nonatomic, readonly) BOOL isScalar;
+
+/**
+ * Returns YES if the value is a boolean.
+ *
+ * @return A BOOL indicating whether the value is a boolean.
+ */
+@property(nonatomic, readonly) BOOL isBoolean;
+
+/**
+ * Returns YES if the value is an integer.
+ *
+ * @return A BOOL indicating whether the value is an integer.
+ */
+@property(nonatomic, readonly) BOOL isInteger;
+
+/**
+ * Returns YES if the value is a double.
+ *
+ * @return A BOOL indicating whether the value is a double.
+ */
+@property(nonatomic, readonly) BOOL isDouble;
+
+/**
+ * Creates an instance encapsulating a Tensor.
+ *
+ * @param value An ExecuTorchTensor instance.
+ * @return A new ExecuTorchValue instance with a tag of
+ * ExecuTorchValueTagTensor.
+ */
++ (instancetype)valueWithTensor:(ExecuTorchTensor *)value
+    NS_SWIFT_NAME(init(_:));
+
+/**
+ * Creates an instance encapsulating a string.
+ *
+ * @param value A string.
+ * @return A new ExecuTorchValue instance with a tag of
+ * ExecuTorchValueTagString.
+ */
++ (instancetype)valueWithString:(ExecuTorchStringValue)value
+    NS_SWIFT_NAME(init(_:));
+
+/**
+ * Creates an instance encapsulating a boolean.
+ *
+ * @param value A boolean.
+ * @return A new ExecuTorchValue instance with a tag of
+ * ExecuTorchValueTagBoolean.
+ */
++ (instancetype)valueWithBoolean:(ExecuTorchBooleanValue)value
+    NS_SWIFT_NAME(init(_:));
+
+/**
+ * Creates an instance encapsulating an integer.
+ *
+ * @param value An integer.
+ * @return A new ExecuTorchValue instance with a tag of
+ * ExecuTorchValueTagInteger.
+ */
++ (instancetype)valueWithInteger:(ExecuTorchIntegerValue)value
+    NS_SWIFT_NAME(init(_:));
+
+/**
+ * Creates an instance encapsulating a double value.
+ *
+ * @param value A double value.
+ * @return A new ExecuTorchValue instance with a tag of
+ * ExecuTorchValueTagDouble.
+ */
++ (instancetype)valueWithDouble:(ExecuTorchDoubleValue)value
+    NS_SWIFT_NAME(init(_:));
+
+/**
+ * Determines whether the current Value is equal to another Value.
+ *
+ * @param other Another ExecuTorchValue instance to compare against.
+ * @return YES if the values have the same tag and equal underlying values;
+ * otherwise, NO.
+ */
+- (BOOL)isEqualToValue:(nullable ExecuTorchValue *)other;
+
+@end
+
+NS_ASSUME_NONNULL_END

package/third-party/ios/ExecutorchLib/frameworks/executorch.xcframework/ios-arm64-simulator/Headers/executorch/extension/module/module.h

@@ -0,0 +1,492 @@
+/*
+ * Copyright (c) Meta Platforms, Inc. and affiliates.
+ * All rights reserved.
+ *
+ * This source code is licensed under the BSD-style license found in the
+ * LICENSE file in the root directory of this source tree.
+ */
+
+#pragma once
+
+#include <memory>
+#include <string>
+#include <unordered_map>
+#include <unordered_set>
+#include <vector>
+
+#include <executorch/runtime/executor/program.h>
+
+namespace executorch {
+namespace extension {
+
+/**
+ * A facade class for loading programs and executing methods within them.
+ */
+class Module {
+public:
+  /**
+   * Enum to define loading behavior.
+   */
+  enum class LoadMode {
+    /// Load the whole file as a buffer.
+    File,
+    /// Use mmap to load pages into memory.
+    Mmap,
+    /// Use memory locking and handle errors.
+    MmapUseMlock,
+    /// Use memory locking and ignore errors.
+    MmapUseMlockIgnoreErrors,
+  };
+
+  /**
+   * Constructs an instance by loading a program from a file with specified
+   * memory locking behavior.
+   *
+   * @param[in] file_path The path to the ExecuTorch program file to load.
+   * @param[in] load_mode The loading mode to use.
+   * @param[in] event_tracer A EventTracer used for tracking and logging events.
+   */
+  explicit Module(const std::string &file_path,
+                  const LoadMode load_mode = LoadMode::MmapUseMlock,
+                  std::unique_ptr<runtime::EventTracer> event_tracer = nullptr);
+
+  /**
+   * Constructs an instance by loading a program from a file with specified
+   * memory locking behavior.
+   *
+   * @param[in] file_path The path to the ExecuTorch program file to load.
+   * @param[in] data_map_path The path to a .ptd file
+   * @param[in] load_mode The loading mode to use.
+   * @param[in] event_tracer A EventTracer used for tracking and logging events.
+   */
+  explicit Module(const std::string &file_path,
+                  const std::string &data_map_path,
+                  const LoadMode load_mode = LoadMode::MmapUseMlock,
+                  std::unique_ptr<runtime::EventTracer> event_tracer = nullptr);
+
+  /**
+   * Constructs an instance with the provided data loader and memory allocator.
+   *
+   * @param[in] data_loader A DataLoader used for loading program data.
+   * @param[in] memory_allocator A MemoryAllocator used for memory management.
+   * @param[in] temp_allocator A MemoryAllocator to use when allocating
+   * temporary data during kernel or delegate execution.
+   * @param[in] event_tracer A EventTracer used for tracking and logging events.
+   * @param[in] data_map_loader A DataLoader used for loading external weights.
+   */
+  explicit Module(
+      std::unique_ptr<runtime::DataLoader> data_loader,
+      std::unique_ptr<runtime::MemoryAllocator> memory_allocator = nullptr,
+      std::unique_ptr<runtime::MemoryAllocator> temp_allocator = nullptr,
+      std::unique_ptr<runtime::EventTracer> event_tracer = nullptr,
+      std::unique_ptr<runtime::DataLoader> data_map_loader = nullptr);
+
+  /**
+   * Constructs an instance using an existing shared program.
+   *
+   * @param[in] program The shared program to use. It's required the data loader
+   * the program uses is valid for the lifetime of the program.
+   * @param[in] memory_allocator A MemoryAllocator used for memory management.
+   * @param[in] temp_allocator A MemoryAllocator to use when allocating
+   * temporary data.
+   * @param[in] event_tracer A EventTracer used for tracking and logging events.
+   * @param[in] data_map_loader A DataLoader used for loading external weights.
+   */
+  explicit Module(
+      std::shared_ptr<runtime::Program> program,
+      std::unique_ptr<runtime::MemoryAllocator> memory_allocator = nullptr,
+      std::unique_ptr<runtime::MemoryAllocator> temp_allocator = nullptr,
+      std::unique_ptr<runtime::EventTracer> event_tracer = nullptr,
+      std::unique_ptr<runtime::DataLoader> data_map_loader = nullptr);
+
+  Module(const Module &) = delete;
+  Module &operator=(const Module &) = delete;
+  Module(Module &&) = delete;
+  Module &operator=(Module &&) = delete;
+
+  /**
+   * Loads the program if needed.
+   *
+   * @param[in] verification The type of verification to do before returning
+   * success.
+   *
+   * @returns An Error to indicate success or failure of the loading process.
+   */
+  ET_NODISCARD
+  runtime::Error load(const runtime::Program::Verification verification =
+                          runtime::Program::Verification::Minimal);
+
+  /**
+   * Checks if the program is loaded.
+   *
+   * @returns true if the program is loaded, false otherwise.
+   */
+  inline bool is_loaded() const { return program_ != nullptr; }
+
+  /**
+   * Get the program. The data loader used by the program is guaranteed to be
+   * valid for the lifetime of the program.
+   *
+   * @returns Shared pointer to the program or nullptr if it's not yet loaded.
+   */
+  inline std::shared_ptr<runtime::Program> program() const { return program_; }
+
+  /**
+   * Get the number of methods available in the loaded program.
+   *
+   * @returns A Result object containing either the number of methods available
+   *          or an error to indicate failure.
+   */
+  runtime::Result<size_t> num_methods();
+
+  /**
+   * Get a list of method names available in the loaded program.
+   * Loads the program and method if needed.
+   *
+   * @returns A set of strings containing the names of the methods, or an error
+   * if the program or method failed to load.
+   */
+  runtime::Result<std::unordered_set<std::string>> method_names();
+
+  /**
+   * Load a specific method from the program and set up memory management if
+   * needed. The loaded method is cached to reuse the next time it's executed.
+   *
+   * @param[in] method_name The name of the method to load.
+   * @param[in] planned_memory The memory-planned buffers to use for mutable
+   * tensor data when executing a method.
+   * @param[in] event_tracer Per-method event tracer to profile/trace methods
+   * individually. When not given, the event tracer passed to the Module
+   * constructor is used. Otherwise, this per-method event tracer takes
+   * precedence.
+   *
+   * @returns An Error to indicate success or failure.
+   */
+  ET_NODISCARD
+  runtime::Error
+  load_method(const std::string &method_name,
+              runtime::HierarchicalAllocator *planned_memory = nullptr,
+              torch::executor::EventTracer *event_tracer = nullptr);
+
+  ET_DEPRECATED ET_NODISCARD runtime::Error inline load_method(
+      const std::string &method_name,
+      torch::executor::EventTracer *event_tracer) {
+    return load_method(method_name, nullptr, event_tracer);
+  }
+
+  /**
+   * Load the 'forward' method from the program and set up memory management if
+   * needed. The loaded method is cached to reuse the next time it's executed.
+   *
+   * @param[in] planned_memory The memory-planned buffers to use for mutable
+   * tensor data when executing the 'forward' method.
+   * @param[in] event_tracer An event tracer used for tracking and logging
+   * events.
+   *
+   * @returns An Error to indicate success or failure.
+   */
+  ET_NODISCARD inline runtime::Error
+  load_forward(runtime::HierarchicalAllocator *planned_memory = nullptr,
+               torch::executor::EventTracer *event_tracer = nullptr) {
+    return load_method("forward", planned_memory, event_tracer);
+  }
+
+  ET_DEPRECATED ET_NODISCARD inline runtime::Error
+  load_forward(torch::executor::EventTracer *event_tracer) {
+    return load_forward(nullptr, event_tracer);
+  }
+
+  /**
+   * Checks if a specific method is loaded.
+   *
+   * @param[in] method_name The name of the method to check.
+   *
+   * @returns true if the method specified by method_name is loaded, false
+   * otherwise.
+   */
+  inline bool is_method_loaded(const std::string &method_name) const {
+    return methods_.count(method_name);
+  }
+
+  /**
+   * Get a method metadata struct by method name.
+   * Loads the program and method if needed.
+   *
+   * @param[in] method_name The name of the method to get the metadata for.
+   *
+   * @returns A method metadata, or an error if the program or method failed to
+   * load.
+   */
+  runtime::Result<runtime::MethodMeta>
+  method_meta(const std::string &method_name);
+
+  /**
+   * Execute a specific method with the given input values and retrieve the
+   * output values. Loads the program and method before executing if needed.
+   *
+   * @param[in] method_name The name of the method to execute.
+   * @param[in] input_values A vector of input values to be passed to the
+   * method.
+   *
+   * @returns A Result object containing either a vector of output values
+   *          from the method or an error to indicate failure.
+   */
+  ET_NODISCARD
+  runtime::Result<std::vector<runtime::EValue>>
+  execute(const std::string &method_name,
+          const std::vector<runtime::EValue> &input_values);
+
+  /**
+   * Execute a specific method with a single input value.
+   * Loads the program and method before executing if needed.
+   *
+   * @param[in] method_name The name of the method to execute.
+   * @param[in] input_value A value to be passed to the method.
+   *
+   * @returns A Result object containing either a vector of output values
+   *          from the method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<std::vector<runtime::EValue>>
+  execute(const std::string &method_name, const runtime::EValue &input_value) {
+    return execute(method_name, std::vector<runtime::EValue>{input_value});
+  }
+
+  /**
+   * Execute a specific method without any input values.
+   * Loads the program and method before executing if needed.
+   *
+   * @param[in] method_name The name of the method to execute.
+   *
+   * @returns A Result object containing either a vector of output values
+   *          from the method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<std::vector<runtime::EValue>>
+  execute(const std::string &method_name) {
+    return execute(method_name, std::vector<runtime::EValue>{});
+  }
+
+  /**
+   * Retrieve the output value of a specific method with the given input values.
+   * Loads the program and method before execution if needed.
+   *
+   * @param[in] method_name The name of the method to execute.
+   * @param[in] input_values A vector of input values to be passed to the
+   * method.
+   *
+   * @returns A Result object containing either the first output value from the
+   * method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<runtime::EValue>
+  get(const std::string &method_name,
+      const std::vector<runtime::EValue> &input_values) {
+    auto result = ET_UNWRAP(execute(method_name, input_values));
+    if (result.empty()) {
+      return runtime::Error::InvalidArgument;
+    }
+    return result[0];
+  }
+
+  /**
+   * Retrieve the output value of a specific method with a single input value.
+   * Loads the program and method before execution if needed.
+   *
+   * @param[in] method_name The name of the method to execute.
+   * @param[in] input_value A value to be passed to the method.
+   *
+   * @returns A Result object containing either the first output value from the
+   * method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<runtime::EValue>
+  get(const std::string &method_name, const runtime::EValue &input_value) {
+    return get(method_name, std::vector<runtime::EValue>{input_value});
+  }
+
+  /**
+   * Retrieve the output value of a specific method without any input values.
+   * Loads the program and method before execution if needed.
+   *
+   * @param[in] method_name The name of the method to execute.
+   *
+   * @returns A Result object containing either the first output value from the
+   * method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<runtime::EValue>
+  get(const std::string &method_name) {
+    return get(method_name, std::vector<runtime::EValue>{});
+  }
+
+  /**
+   * Execute the 'forward' method with the given input values and retrieve the
+   * output values. Loads the program and method before executing if needed.
+   *
+   * @param[in] input_values A vector of input values for the 'forward' method.
+   *
+   * @returns A Result object containing either a vector of output values
+   *          from the 'forward' method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<std::vector<runtime::EValue>>
+  forward(const std::vector<runtime::EValue> &input_values) {
+    return execute("forward", input_values);
+  }
+
+  /**
+   * Execute the 'forward' method with a single value.
+   * Loads the program and method before executing if needed.
+   *
+   * @param[in] input_value A value for the 'forward' method.
+   *
+   * @returns A Result object containing either a vector of output values
+   *          from the 'forward' method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<std::vector<runtime::EValue>>
+  forward(const runtime::EValue &input_value) {
+    return forward(std::vector<runtime::EValue>{input_value});
+  }
+
+  /**
+   * Execute the 'forward' method without any input values.
+   * Loads the program and method before executing if needed.
+   *
+   * @returns A Result object containing either a vector of output values
+   *          from the 'forward' method or an error to indicate failure.
+   */
+  ET_NODISCARD inline runtime::Result<std::vector<runtime::EValue>> forward() {
+    return forward(std::vector<runtime::EValue>{});
+  }
+
+  /**
+   * Sets a single input value for a specific method.
+   *
+   * @param[in] method_name The name of the method.
+   * @param[in] input_value The EValue to set as the method input.
+   * @param[in] input_index Zero-based index of the input to set.
+   *
+   * @returns An Error to indicate success or failure.
+   */
+  ET_NODISCARD
+  runtime::Error set_input(const std::string &method_name,
+                           const runtime::EValue &input_value,
+                           size_t input_index);
+
+  /**
+   * Sets a single input value for the "forward" method.
+   *
+   * @param[in] input_value The EValue to set as the method input.
+   * @param[in] input_index Zero-based index of the input to set.
+   *
+   * @returns An Error to indicate success or failure.
+   */
+  ET_NODISCARD
+  inline runtime::Error set_input(const runtime::EValue &input_value,
+                                  size_t input_index) {
+    return set_input("forward", input_value, input_index);
+  }
+
+  /**
+   * Sets all input values for a specific method.
+   *
+   * @param[in] method_name The name of the method.
+   * @param[in] input_values A vector of EValues to set as the method inputs.
+   *
+   * @returns An Error to indicate success or failure.
+   */
+  ET_NODISCARD
+  runtime::Error set_inputs(const std::string &method_name,
+                            const std::vector<runtime::EValue> &input_values);
+
+  /**
+   * Sets all input values for the "forward" method.
+   *
+   * @param[in] input_values A vector of EValues to set as the method inputs.
+   *
+   * @returns An Error to indicate success or failure.
+   */
+  ET_NODISCARD
+  inline runtime::Error
+  set_inputs(const std::vector<runtime::EValue> &input_values) {
+    return set_inputs("forward", input_values);
+  }
+
+  /**
+   * Sets the output tensor for a specific method.
+   *
+   * @param[in] method_name The name of the method.
+   * @param[in] output_value The EValue containing the Tensor to set as the
+   * method output.
+   * @param[in] output_index Zero-based index of the output to set.
+   *
+   * @returns An Error to indicate success or failure.
+   *
+   * @note Only Tensor outputs are currently supported for setting.
+   */
+  ET_NODISCARD
+  runtime::Error set_output(const std::string &method_name,
+                            runtime::EValue output_value,
+                            size_t output_index = 0);
+
+  /**
+   * Sets the output tensor for the "forward" method.
+   *
+   * @param[in] output_value The EValue containing the Tensor to set as the
+   * method output.
+   * @param[in] output_index Zero-based index of the output to set.
+   *
+   * @returns An Error to indicate success or failure.
+   *
+   * @note Only Tensor outputs are currently supported for setting.
+   */
+  ET_NODISCARD
+  inline runtime::Error set_output(runtime::EValue output_value,
+                                   size_t output_index = 0) {
+    return set_output("forward", std::move(output_value), output_index);
+  }
+
+  /**
+   * Retrieves the EventTracer instance being used by the Module.
+   * EventTracer is used for tracking and logging events during the execution
+   * of methods.
+   *
+   * @returns A pointer to the EventTracer instance. Returns nullptr if no
+   * EventTracer is set.
+   */
+  inline runtime::EventTracer *event_tracer() const {
+    return event_tracer_.get();
+  }
+
+private:
+  struct MethodHolder {
+    std::vector<std::vector<uint8_t>> planned_buffers;
+    std::vector<runtime::Span<uint8_t>> planned_spans;
+    std::unique_ptr<runtime::HierarchicalAllocator> planned_memory;
+    std::unique_ptr<runtime::MemoryManager> memory_manager;
+    std::unique_ptr<runtime::Method> method;
+    std::vector<runtime::EValue> inputs;
+  };
+
+  std::string file_path_;
+  std::string data_map_path_;
+  LoadMode load_mode_{LoadMode::MmapUseMlock};
+  std::shared_ptr<runtime::Program> program_;
+  std::unique_ptr<runtime::DataLoader> data_loader_;
+  std::unique_ptr<runtime::MemoryAllocator> memory_allocator_;
+  std::unique_ptr<runtime::MemoryAllocator> temp_allocator_;
+  std::unique_ptr<runtime::EventTracer> event_tracer_;
+  std::unique_ptr<runtime::DataLoader> data_map_loader_;
+  std::unique_ptr<runtime::NamedDataMap> data_map_;
+
+protected:
+  std::unordered_map<std::string, MethodHolder> methods_;
+
+  friend class ExecuTorchJni;
+};
+
+} // namespace extension
+} // namespace executorch
+
+namespace torch {
+namespace executor {
+// TODO(T197294990): Remove these deprecated aliases once all users have moved
+// to the new `::executorch` namespaces.
+using ::executorch::extension::Module;
+} // namespace executor
+} // namespace torch