Add some missing operators (#1524)

nnvm/python/nnvm/top/tensor.py

@@ -156,6 +156,14 @@ reg.register_schedule("elemwise_mul", _fschedule_broadcast)
 reg.register_pattern("elemwise_div", OpPattern.BROADCAST)
 reg.register_schedule("elemwise_div", _fschedule_broadcast)
 
+# elemwise_mod
+reg.register_pattern("elemwise_mod", OpPattern.BROADCAST)
+reg.register_schedule("elemwise_mod", _fschedule_broadcast)
+
+# elemwise_pow
+reg.register_pattern("elemwise_pow", OpPattern.BROADCAST)
+reg.register_schedule("elemwise_pow", _fschedule_broadcast)
+
 # broadcast_add
 reg.register_pattern("broadcast_add", OpPattern.BROADCAST)
 reg.register_schedule("broadcast_add", _fschedule_broadcast)

nnvm/src/top/tensor/elemwise.cc

@@ -337,6 +337,30 @@ NNVM_REGISTER_ELEMWISE_BINARY_OP(elemwise_div)
     };
 });
 
+NNVM_REGISTER_ELEMWISE_BINARY_OP(elemwise_mod)
+  .describe(R"code(Element-wise modulo
+
+)code" NNVM_ADD_FILELINE)
+.set_support_level(1)
+.set_attr<FTVMCompute>(
+  "FTVMCompute", [](const NodeAttrs& attrs,
+                    const Array<Tensor>& inputs,
+                    const Array<Tensor>& out_info) {
+      return Array<Tensor>{ topi::mod(inputs[0], inputs[1]) };
+});
+
+NNVM_REGISTER_ELEMWISE_BINARY_OP(elemwise_pow)
+  .describe(R"code(Element-wise power
+
+)code" NNVM_ADD_FILELINE)
+.set_support_level(1)
+.set_attr<FTVMCompute>(
+  "FTVMCompute", [](const NodeAttrs& attrs,
+                    const Array<Tensor>& inputs,
+                    const Array<Tensor>& out_info) {
+      return Array<Tensor>{ topi::power(inputs[0], inputs[1]) };
+});
+
 // negative
 NNVM_REGISTER_ELEMWISE_UNARY_OP(negative)
 .describe(R"code(Elemenwise numeric negative