forked from jiuyuan/InfiniTensor
![]() * 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 | ||
kernels/cuda | ||
nnet | ||
operators | ||
script |