1. update Spread, ReduceSumAll, Multiply, Unsqueeze, Softmax, CrossEntropy with float16 datatype 2. modify the implementation of fnnlm to support float16 computation, but it remains some bugs and the loss is nan
| Name |
Last commit
|
Last Update |
|---|---|---|
| .. | ||
| ReduceMax.cpp | 正在载入提交数据... | |
| ReduceMax.cu | 正在载入提交数据... | |
| ReduceMax.cuh | 正在载入提交数据... | |
| ReduceMax.h | 正在载入提交数据... | |
| ReduceMean.cpp | 正在载入提交数据... | |
| ReduceMean.h | 正在载入提交数据... | |
| ReduceStandardVariance.h | 正在载入提交数据... | |
| ReduceSum.cpp | 正在载入提交数据... | |
| ReduceSum.cu | 正在载入提交数据... | |
| ReduceSum.cuh | 正在载入提交数据... | |
| ReduceSum.h | 正在载入提交数据... | |
| ReduceSumAll.cpp | 正在载入提交数据... | |
| ReduceSumAll.h | 正在载入提交数据... | |
| ReduceSumSquared.cpp | 正在载入提交数据... | |
| ReduceSumSquared.h | 正在载入提交数据... | |
| ReduceVariance.cpp | 正在载入提交数据... | |
| ReduceVariance.h | 正在载入提交数据... |