Refactor OperatorImpl for backend/export
Compare changes
Files
32@@ -27,30 +27,33 @@
@@ -27,30 +27,33 @@
template <class T> void forward_(const std::vector<Tensor>& inputs, const std::vector<std::vector<int>>& inputsDims, const std::vector<std::vector<int>>& inputsStrides);
template <class T> void backward_(const Tensor& outGrad, const std::vector<std::vector<int>>& inputsDims, const std::vector<std::vector<int>>& inputsStrides);