lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix transpose_op

This commit is contained in:
guoye yang 2020-12-15 17:28:12 +08:00
parent 73eb05b36e
commit f9be78c1d4
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/ops/transpose_op.cc
View File

@ -14,11 +14,6 @@ namespace jittor {
static auto make_transpose = get_op_info("transpose") static auto make_transpose = get_op_info("transpose")
.get_constructor<VarPtr, Var*, NanoVector>(); .get_constructor<VarPtr, Var*, NanoVector>();
#ifdef HAS_CUDA
static auto make_reshape = get_op_info("reshape")
.get_constructor<VarPtr, Var*, NanoVector>();
#endif
TransposeOp::TransposeOp(Var* x, NanoVector axes_) : x(x), axes(axes_) { TransposeOp::TransposeOp(Var* x, NanoVector axes_) : x(x), axes(axes_) {
int i=0; int i=0;
for (; i<axes.size(); i++) for (; i<axes.size(); i++)