nn-bench-master.zip
(预估有个40文件)
nn-bench-master
.gitignore
0B
nns
.gitignore
33B
__init__.py
0B
nnstats.py
11KB
test.py
1019B
nnutils.py
8KB
modules
elements.py
6KB
__init__.py
87B
flops.py
11KB
macs.py
7KB
memory.py
8KB
test
basic_kernel.py
1KB
cnn.py
2KB
image_classification
.gitignore
7B
script
npu_backward.sh
771B
npu_env.sh
1KB
npu_forward.sh
737B
gpu_forward.sh
688B
gpu_backward.sh
758B
pytorch
.gitignore
49B
vision.py
6KB
basic-kernels
.gitignore
7B
analysis
npu_linear.json
331B
get_perf.py
2KB
script
gpu_conv2d.sh
87B
npu_env.sh
1KB
npu_conv2d.sh
154B
gpu_linear.sh
909B
gpu_nsight_conv2d.sh
2KB
npu_linear.sh
941B
npu_rnn.sh
1KB
pytorch
.gitignore
49B
conv2d.py
5KB
rnn.py
5KB
linear.py
5KB
seq2seq
.gitignore
70B
pytorch
seq2seq.py
2KB
decoder.py
1011B
encoder.py
601B
train.py
6KB
暂无评论