Learning Metrics operators support
Compare changes
Files
8+ 62
− 0
\ No newline at end of file
To Support Adam Optimizer, this MR adds operator Sqrt's forward implementation. It also fixes ReduceMean and ReduceSum in case no reduction is needed (encountered in BCE and MSE unit tests)
SqrtImpl.hpp
, SqrtImpl.cpp
, SqrtImpl_CUDA_kernels.hpp
, SqrtImpl_CUDA_kernels.cu
and Test_SqrtImpl.cpp
, add Sqrt forward and backward operation;ReduceMeanImpl.cpp
, fix ReduceMean forward;ReduceSumImpl.cpp
, fix ReduceSum forward;Copyright © Eclipse Foundation, Inc. All Rights Reserved. Privacy Policy | Terms of Use | Copyright Agent