1. add gather function 2. add cross entropy forward computation and backward computation 3. code optimization 4. merge with xiaotong-working branch
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 | 正在载入提交数据... |