remove tensor.h and declare Prod in pattern_utils.h

apache · Feb 6, 2021 · 9877ddf · 9877ddf
1 parent 9f462d7
commit 9877ddf
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 42 deletions.
diff --git a/src/relay/op/tensor/reduce.cc b/src/relay/op/tensor/reduce.cc
@@ -475,10 +475,6 @@ Array<te::Tensor> ProdCompute(const Attrs& attrs, const Array<te::Tensor>& input
   return ReduceCompute(attrs, inputs, out_type, topi::prod);
 }
 
-Expr Prod(Expr data, Array<Integer> axis, bool keepdims, bool exclude) {
-  return MakeReduce(data, axis, keepdims, exclude, "prod");
-}
-
 TVM_REGISTER_GLOBAL("relay.op._make.prod").set_body_typed(Prod);
 
 RELAY_REGISTER_OP("prod")

diff --git a/src/relay/op/tensor/tensor.h b/src/relay/op/tensor/tensor.h
diff --git a/src/relay/op/vm/vm.h b/src/relay/op/vm/vm.h
@@ -33,6 +33,7 @@ Expr InvokeTVMOp(Expr func, Expr inputs, Expr outputs);
 Expr ShapeFunc(Expr func, Expr inputs, Expr outputs, Array<tvm::Integer> is_input);
 Expr ShapeOf(Expr expr);
 Expr ReshapeTensor(Expr data, Expr shape, Array<PrimExpr> newshape);
+
 }  // namespace relay
 }  // namespace tvm
 

diff --git a/src/relay/transforms/memory_alloc.cc b/src/relay/transforms/memory_alloc.cc
@@ -42,7 +42,6 @@
 
 #include "../backend/compile_engine.h"
 #include "../op/memory/memory.h"
-#include "../op/tensor/tensor.h"
 #include "../op/vm/vm.h"
 #include "let_list.h"
 #include "pattern_utils.h"

diff --git a/src/relay/transforms/pattern_utils.h b/src/relay/transforms/pattern_utils.h
@@ -644,6 +644,10 @@ static inline Expr Sum(Expr data, Array<Integer> axis, bool keepdims, bool exclu
   return MakeReduce(data, axis, keepdims, exclude, "sum");
 }
 
+static inline Expr Prod(Expr data, Array<Integer> axis, bool keepdims, bool exclude) {
+  return MakeReduce(data, axis, keepdims, exclude, "prod");
+}
+
 static inline Expr Reshape(Expr data, Array<Integer> newshape) {
   return MakeReshape(data, newshape);
 }