return optional in outputcharacteristic

microsoft · Oct 26, 2023 · 04ff3ec · 04ff3ec
1 parent 06d0ff3
commit 04ff3ec
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyop/pykernel.h b/pyop/pykernel.h
@@ -117,7 +117,8 @@ struct PyCustomOpFactory : public OrtCustomOp {
     };
 
     OrtCustomOp::GetOutputCharacteristic = [](const OrtCustomOp* this_, size_t index) noexcept {
-      return OrtCustomOpInputOutputCharacteristic::INPUT_OUTPUT_REQUIRED;
+      return index == 0 ? OrtCustomOpInputOutputCharacteristic::INPUT_OUTPUT_REQUIRED
+                    : OrtCustomOpInputOutputCharacteristic::INPUT_OUTPUT_OPTIONAL;
     };
   }