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 |
||
---|---|---|
.. | ||
core | ||
kernels | ||
nnet | ||
operators | ||
script |