wangwei
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
2253 Commits
28 Branches
25 MB
551 Download
 
 
 
 
 
 
Tree: cbbca5fb10
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'cbbca5fb10'
${ noResults }
MegEngine/src/opr/impl/dnn
Megvii Engine Team cbbca5fb10 feat(mge): add softmax op use cudnn api 4 years ago
..
adaptive_pooling.cpp style(all): reformat c++ code 4 years ago
batch_norm.cpp style(all): reformat c++ code 4 years ago
convolution.cpp style(all): reformat c++ code 4 years ago
correlation.cpp style(all): reformat c++ code 4 years ago
dnn.oprdecl feat(mge): add softmax op use cudnn api 4 years ago
dnn.sereg.h feat(mge): add softmax op use cudnn api 4 years ago
fake_quant.cpp style(all): reformat c++ code 4 years ago
helper.h style(all): reformat c++ code 4 years ago
images2neibs.cpp style(all): reformat c++ code 4 years ago
layer_norm.cpp feat(mgb/opr): add layernorm forward and backward kernel 4 years ago
local.cpp style(all): reformat c++ code 4 years ago
lrn.cpp style(all): reformat c++ code 4 years ago
lsq.cpp style(all): reformat c++ code 4 years ago
pooling.cpp fix(mgb): fix loss execution policy after opr shallow copy 4 years ago
rnn.cpp fix(dnn): add naive lstm kernel 4 years ago
roi_align.cpp style(all): reformat c++ code 4 years ago
roi_pooling.cpp style(all): reformat c++ code 4 years ago
sliding_window_transpose.cpp style(all): reformat c++ code 4 years ago
softmax.cpp feat(mge): add softmax op use cudnn api 4 years ago
tqt.cpp style(all): reformat c++ code 4 years ago
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境，不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序，请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉，欢迎访问 MegStudio 平台

C++ Cuda Python C SVG other