Add conf axis for bias_add for any axis channel (#2093)
* Fix tick in multi node parallel (#2042) * check in fixes * fix by adding boxing method * register tick op * move code and add more check * fix typo * fix bug when filtering op nodes before adding tick * fix wheel build not adding .so (#2052) * color plan dot VERSION-2 (#2045) * bias_add completion * follow comment * make conf axis required
Showing
- oneflow/core/job_completer/bias_add_grad.cpp 8 additions, 2 deletionsoneflow/core/job_completer/bias_add_grad.cpp
- oneflow/core/kernel/bias_add_kernel.cpp 28 additions, 7 deletionsoneflow/core/kernel/bias_add_kernel.cpp
- oneflow/core/kernel/bias_add_kernel.cu 52 additions, 0 deletionsoneflow/core/kernel/bias_add_kernel.cu
- oneflow/core/kernel/bias_add_kernel.h 12 additions, 0 deletionsoneflow/core/kernel/bias_add_kernel.h
- oneflow/core/operator/bias_add_op.cpp 5 additions, 2 deletionsoneflow/core/operator/bias_add_op.cpp
- oneflow/core/operator/op_conf.proto 2 additions, 0 deletionsoneflow/core/operator/op_conf.proto
Please register or sign in to comment