integration

Author: kirklandsign

extension/android/jni/jni_layer.cpp

@@ -354,12 +354,12 @@ class ExecuTorchJni : public facebook::jni::HybridClass<ExecuTorchJni> {
   }
 
   jint forward_ones() {
+    auto&& load_result = module_->load_method("forward");
     auto&& buf = prepare_input_tensors(*(module_->methods_["forward"].method));
-    auto&& result = module_->forward();
-    return (jint) result.error();
+    auto&& result = module_->methods_["forward"].method->execute();
+    return (jint) result;
   }
 
-
   static void registerNatives() {
     registerHybrid({
         makeNativeMethod("initHybrid", ExecuTorchJni::initHybrid),

extension/android/src/main/java/org/pytorch/executorch/Module.java

@@ -79,6 +79,12 @@ public static Module load(final String modelPath) {
    * @return return value from the 'forward' method.
    */
   public EValue[] forward(EValue... inputs) {
+    if (inputs.length == 0) {
+      // forward default args (ones)
+      mNativePeer.forwardOnes();
+      // discard the return value
+      return null;
+    }
     return mNativePeer.forward(inputs);
   }