Skip to content

Commit 912dfa9

Browse files
committed
fix: update ExecutorchModule header
1 parent 0667c20 commit 912dfa9

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

packages/react-native-executorch/common/rnexecutorch/bindings/ExecutorchModule.h

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,16 +5,20 @@
55
#include <ReactCommon/CallInvoker.h>
66
#include <executorch/extension/module/module.h>
77
#include <jsi/jsi.h>
8+
#include <rnexecutorch/jsi/OwningArrayBuffer.h>
89
#include <rnexecutorch/utils/JsiTensorView.h>
910

11+
using namespace facebook;
12+
1013
namespace rnexecutorch {
1114

1215
class ExecutorchModule {
1316
public:
1417
ExecutorchModule(const std::string &modelSource,
1518
std::shared_ptr<facebook::react::CallInvoker> callInvoker);
1619
std::vector<int32_t> getInputShape(std::string method_name, int index);
17-
int forward(std::vector<JsiTensorView> tensorViewVec);
20+
std::vector<std::shared_ptr<OwningArrayBuffer>>
21+
forward(std::vector<JsiTensorView> tensorViewVec);
1822

1923
protected:
2024
std::unique_ptr<executorch::extension::Module> module;

0 commit comments

Comments
 (0)