Dev opt split axis (#2121) (39b47855) · Commits · Summer2021 / 210130133 · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

2

210130133
- Activity
- Members
- Labels
- Wiki
- Releases
- Package Registry
- Terraform modules

Snippets Groups Projects

39b47855

Commit 39b47855 authored 5 years ago by

Li Xinqi Committed by cheng cheng 5 years ago

Downloads
- Patches
- Plain Diff

Dev opt split axis (#2121)

* optional split_axis

* backup

* VariableConf::(OptInt64 split_axis)

* backup

parent c206bd70

No related branches found

No related tags found

Changes 13

Hide whitespace changes

Inline Side-by-side

Showing

oneflow/core/job_completer/adam_optm.cpp 7 additions, 16 deletions

oneflow/core/job_completer/adam_optm.cpp
oneflow/core/job_completer/auto_global_step.cpp 1 addition, 0 deletions

oneflow/core/job_completer/auto_global_step.cpp
oneflow/core/job_completer/lars_optm.cpp 0 additions, 1 deletion

oneflow/core/job_completer/lars_optm.cpp
oneflow/core/job_completer/momentum_optm.cpp 0 additions, 1 deletion

oneflow/core/job_completer/momentum_optm.cpp
oneflow/core/job_completer/optimizer.cpp 0 additions, 6 deletions

oneflow/core/job_completer/optimizer.cpp
oneflow/core/job_completer/optimizer.h 0 additions, 2 deletions

oneflow/core/job_completer/optimizer.h
oneflow/core/operator/op_conf.proto 1 addition, 1 deletion

oneflow/core/operator/op_conf.proto
oneflow/core/operator/variable_op.cpp 33 additions, 21 deletions

oneflow/core/operator/variable_op.cpp
oneflow/python/framework/input_blob_def.py 2 additions, 2 deletions

oneflow/python/framework/input_blob_def.py
oneflow/python/ops/get_variable.py 6 additions, 3 deletions

oneflow/python/ops/get_variable.py
oneflow/python/ops/losses/__init__.py 0 additions, 0 deletions

oneflow/python/ops/losses/__init__.py
oneflow/python/ops/losses/add_loss.py 9 additions, 0 deletions

oneflow/python/ops/losses/add_loss.py
oneflow/python/ops/nn_ops.py 1 addition, 0 deletions

oneflow/python/ops/nn_ops.py

with 60 additions and 53 deletions

0% or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment