forked from jiuyuan/InfiniTensor
48847958d0
* impl sqrt on CUDA fix parser of Gather and ReduceMean * fix test_gather * fix test_cuda_gather * impl sqrt cpu and add sqrt to test_cuda_unary * cuda_unary supports arbitary shapes * fix SplitOp with dim=-1 * fix SplitOp with dim=-1 |
||
|---|---|---|
| .. | ||
| test_batch_norm.cc | ||
| test_clip.cc | ||
| test_concat.cc | ||
| test_conv.cc | ||
| test_conv_transposed_2d.cc | ||
| test_element_wise.cc | ||
| test_extend.cc | ||
| test_gather.cc | ||
| test_matmul.cc | ||
| test_pad.cc | ||
| test_pooling.cc | ||
| test_reduce_mean.cc | ||
| test_reshape.cc | ||
| test_resize.cc | ||
| test_slice.cc | ||
| test_split.cc | ||
| test_transpose.cc | ||