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

Update arg_reduce_op.h

This commit is contained in:
Jittor 2021-03-16 18:45:47 +08:00 committed by GitHub
parent 8b987b928e
commit 80ad0d81ae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/ops/arg_reduce_op.h
View File

@ -28,7 +28,7 @@ struct ArgReduceOp : Op {
* [in] op: "max" or "min".
* [in] dim: int. Specifies which dimension to reduce.
* [in] dim: int. Specifies which dimension to be reduced.
* [in] keepdim: bool. Whether the output has ``dim`` retained or not.
@ -54,4 +54,4 @@ struct ArgReduceOp : Op {
DECLARE_jit_run;
};
} // jittor
} // jittor