Skip to content
Snippets Groups Projects
Unverified Commit de840ff4 authored by Yao Chi's avatar Yao Chi Committed by GitHub
Browse files

Add math ops modules (#4629)


* add modules:

- sum
- mean
- add
- sub
- mul
- div
- reciprocal

* add +, -, *, / for tensor

* add test case for add,div,mean,mul,reciprocal,sub,sum

* refine test cases

* refine sum module

* remove useless oneflow_export for mean and sum

* add tensor op test for mean and sum

* add CrossEntropyLoss module

* refine Sum

* sum axis attr not found

* refine sum

* refine math ops

* refien cross entropy loss

* refine (not finished)

* fix scalar div, sub

* rename test Module name and refine test case

* rename test Module name and refine test case

* add neg method for Tensor

* remove useless improt items

* refine

* reformat

* remove useless dynamic attr

* fix math_ops and loss

* rename test

* fix format

* delete name

* refine codes and format

* style(*): format codes

* fix(MathOp): fix merge bug

Co-authored-by: default avatarsimonJJJ <821898965@qq.com>
Co-authored-by: default avatarwyg1997 <wyg19970408@gmail.com>
Co-authored-by: default avataroneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
parent 2ae5176d
No related branches found
No related tags found
No related merge requests found
Showing
with 1156 additions and 24 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment