forked from jiuyuan/InfiniTensor
9032cbb973
* ADD:reshape/flatten/identity operators and cuda kernel. fix: use cudaMemcpyAsync clang format. ADD flatten/identity operator. add test for reshape. ADD: reshape operator and cuda kernel. * Fix: seperate CUDA tests & remove old header Co-authored-by: Liyan Zheng <liyan-zheng@outlook.com> |
||
---|---|---|
.. | ||
core | ||
cuda | ||
nnet | ||
operators | ||
utils | ||
test.h |