feat(opr): add confidential operator

GitOrigin-RevId: 53c2d4bc45

imperative/python/megengine/functional/nn.py

@@ -1579,3 +1579,9 @@ def batched_nms(
     indices = indices[0][: count.item()]
     keep_inds = sorted_idx[indices]
     return keep_inds
+
+
+
+
+from .loss import *  # isort:skip
+from .quantized import conv_bias_activation  # isort:skip

imperative/python/test/unit/functional/test_functional.py

@@ -551,3 +551,5 @@ def test_nms_is_same():
     assert op1 != op3
     assert op1 != op4
     assert op3 != op4
+
+

src/opr/impl/misc.cpp

@@ -159,6 +159,7 @@ void Cumsum::init_output_static_infer_desc() {
             {SourceType::DEP, {{input(0), DepType::SHAPE}}, infer_workspace});
 }
 
+
 /* ================= CondTake =================  */
 MGB_DYN_TYPE_OBJ_FINAL_IMPL(CondTake);
 

src/opr/impl/misc.oprdecl

@@ -63,4 +63,5 @@ decl_opr('TopK',
          inputs=['data', 'k'], params='TopK',
          desc='Select the top k values from sorted result.')
 
+
 # vim: ft=python

src/opr/impl/misc.sereg.h

@@ -70,6 +70,7 @@ namespace opr {
     using CumsumV1 = opr::Cumsum;
     MGB_SEREG_OPR(CumsumV1, 1);
 
+
 } // namespace opr
 } // namespace mgb
 

src/opr/include/megbrain/opr/misc.h

@@ -94,6 +94,7 @@ MGB_DEFINE_OPR_CLASS(Cumsum, cg::SingleCNOperatorNodeBaseT<
         void init_output_static_infer_desc() override;
 };
 
+
 namespace intl {
 using CondTakeBase =
         cg::SingleCNOperatorNode<cg::OperatorNodeBase,

src/serialization/impl/schema.fbs

@@ -28,6 +28,7 @@ table Blob {
 }
 
 table Reserved0 {}
+table Reserved1 {}
 
 union OperatorParam {
     param.Empty = 1,
@@ -100,6 +101,7 @@ union OperatorParam {
     param.Remap = 68,
     param.NMSKeep = 69,
     param.AdaptivePooling = 70,
+    Reserved1 = 71,
 }
 
 table Operator {

tools/param_defs/mgb_opr_param_defs.py

@@ -143,3 +143,4 @@ pdef('PersistentOutputStorage').add_fields(
                ' no branch is taken')
            )
  )
+