Dev l2 norm (#1567)
* dev l2 normalize * fix no newline at the end of file * refine by review * little refine * refine by review * rename * replace input of output in backward * compatible with TensorFlow * int64_t -> int32_t * little refine
Showing
- oneflow/core/kernel/l2_normalize_kernel.cpp 85 additions, 0 deletionsoneflow/core/kernel/l2_normalize_kernel.cpp
- oneflow/core/kernel/l2_normalize_kernel.cu 103 additions, 0 deletionsoneflow/core/kernel/l2_normalize_kernel.cu
- oneflow/core/kernel/l2_normalize_kernel.h 33 additions, 0 deletionsoneflow/core/kernel/l2_normalize_kernel.h
- oneflow/core/operator/l2_normalize_op.cpp 31 additions, 0 deletionsoneflow/core/operator/l2_normalize_op.cpp
- oneflow/core/operator/l2_normalize_op.h 23 additions, 0 deletionsoneflow/core/operator/l2_normalize_op.h
- oneflow/core/operator/op_conf.proto 8 additions, 0 deletionsoneflow/core/operator/op_conf.proto
Please register or sign in to comment