[Feature](mluOpLogcumsumexp): new parameters (#96)

Cambricon · Jun 27, 2024 · 8a4bd1f · 8a4bd1f
1 parent 422efa2
commit 8a4bd1f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/mlu_op_test.proto b/mlu_op_test.proto
@@ -426,6 +426,7 @@ message Node {
   optional BatchMatMulBCastParam batch_matmul_bcast_param       = 132;   // param
   optional FFTParam fft_param = 65536; // param
   optional DCNParam dcn_param       = 405;
+  optional LogcumsumexpParam logcumsumexp_param       = 19980423;
 }
 
 
@@ -1177,3 +1178,8 @@ message DCNParam {
   optional int32 im2col_step = 7;
   optional DataType compute_type = 8;
 }
+
+// param to call mluOpLogcumsumexp()
+message LogcumsumexpParam {
+  optional int32 dim = 1 [default = 0];
+}