Refactor RandomGenerator (#900)
* Refactor random generator * refactor rng * remove random_generator_.reset if branch * combine macro and remove const * add declaration for generalized rng * remove generalized rng definition
Showing
- oneflow/core/actor/actor.cpp 2 additions, 2 deletionsoneflow/core/actor/actor.cpp
- oneflow/core/kernel/dropout_kernel.cpp 4 additions, 3 deletionsoneflow/core/kernel/dropout_kernel.cpp
- oneflow/core/kernel/dropout_kernel.h 3 additions, 2 deletionsoneflow/core/kernel/dropout_kernel.h
- oneflow/core/kernel/kernel.cpp 5 additions, 4 deletionsoneflow/core/kernel/kernel.cpp
- oneflow/core/kernel/kernel.h 6 additions, 2 deletionsoneflow/core/kernel/kernel.h
- oneflow/core/kernel/opkernel_test_case.cpp 1 addition, 1 deletiononeflow/core/kernel/opkernel_test_case.cpp
- oneflow/core/kernel/random_generator.cpp 17 additions, 34 deletionsoneflow/core/kernel/random_generator.cpp
- oneflow/core/kernel/random_generator.cu 17 additions, 13 deletionsoneflow/core/kernel/random_generator.cu
- oneflow/core/kernel/random_generator.h 20 additions, 21 deletionsoneflow/core/kernel/random_generator.h
Please register or sign in to comment