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> |
||
---|---|---|
.. | ||
test_conv.cc | ||
test_conv_transposed_2d.cc | ||
test_element_wise.cc | ||
test_matmul.cc | ||
test_pooling.cc | ||
test_reshape.cc |