File tree Expand file tree Collapse file tree
packages/react-native-executorch/common/rnexecutorch/models Expand file tree Collapse file tree Original file line number Diff line number Diff line change 88#include < jsi/jsi.h>
99#include < rnexecutorch/host_objects/JSTensorViewIn.h>
1010#include < rnexecutorch/host_objects/JSTensorViewOut.h>
11- #include < rnexecutorch/jsi/OwningArrayBuffer.h>
1211
1312namespace rnexecutorch {
1413using namespace facebook ;
@@ -23,7 +22,6 @@ class BaseModel {
2322 std::vector<int32_t > getInputShape (std::string method_name, int index);
2423 std::vector<std::vector<int32_t >>
2524 getAllInputShapes (std::string methodName = " forward" );
26-
2725 std::vector<std::shared_ptr<JSTensorViewOut>>
2826 forwardJS (std::vector<JSTensorViewIn> tensorViewVec);
2927
@@ -35,9 +33,9 @@ class BaseModel {
3533 // (unnecessary copies instead of working on JS memory). In this case
3634 // CallInvoker can be used to get jsi::Runtime, and use it in a safe manner.
3735 std::shared_ptr<react::CallInvoker> callInvoker;
38- std::size_t memorySizeLowerBound{0 };
3936
4037private:
38+ std::size_t memorySizeLowerBound{0 };
4139 std::unique_ptr<executorch::extension::Module> module ;
4240 std::vector<int32_t > getTensorShape (const executorch::aten::Tensor &tensor);
4341};
You can’t perform that action at this time.
0 commit comments