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.
2614 Commits
28 Branches
25 MB
582 Download
 
 
 
 
 
 
Tree: e7587617c2
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'e7587617c2'
${ noResults }
MegEngine/imperative/src/impl/ops
Megvii Engine Team 1192a9a61b fix(imperative): fix adaptive pool2d 3 years ago
..
adaptive_pooling.cpp fix(imperative): fix adaptive pool2d 3 years ago
atlas_runtime.cpp chore(license): remove all license in file header 3 years ago
autogen.cpp chore(license): remove all license in file header 3 years ago
backward_graph.cpp chore(license): remove all license in file header 3 years ago
batch_norm.cpp chore(license): remove all license in file header 3 years ago
broadcast.cpp chore(license): remove all license in file header 3 years ago
cambricon_runtime.cpp chore(license): remove all license in file header 3 years ago
collective_comm.cpp chore(license): remove all license in file header 3 years ago
cond_take.cpp chore(license): remove all license in file header 3 years ago
convolution.cpp fix(imperative/ops): fix the vmemory problem in 1.9 3 years ago
custom_opdef.cpp chore(license): remove all license in file header 3 years ago
deformable_conv2d.cpp chore(license): remove all license in file header 3 years ago
deformable_psroi_pooling.cpp chore(license): remove all license in file header 3 years ago
elemwise.cpp chore(license): remove all license in file header 3 years ago
extern_opr.cpp chore(license): remove all license in file header 3 years ago
indexing.cpp fix(imperative): make functional ops support negative axis 3 years ago
io_remote.cpp chore(license): remove all license in file header 3 years ago
lamb.cpp feat(mge/dnn): add lamb optimizer 3 years ago
layer_norm.cpp refactor(ops/layer_norm): speed up the host speed of layer_norm 3 years ago
magicmind_runtime.cpp chore(license): remove all license in file header 3 years ago
matmul.cpp perf(mge): override grad of matmul 3 years ago
matrix_inverse.cpp chore(license): remove all license in file header 3 years ago
misc.cpp chore(license): remove all license in file header 3 years ago
nms.cpp chore(license): remove all license in file header 3 years ago
opr_attr.cpp chore(license): remove all license in file header 3 years ago
pixel_shuffle.cpp perf(mge): speed up PixelShuffle 3 years ago
pooling.cpp chore(license): remove all license in file header 3 years ago
reduce.cpp fix(imperative): fix buildin reduce keepdim 3 years ago
resize.cpp chore(license): remove all license in file header 3 years ago
rng.cpp chore(license): remove all license in file header 3 years ago
rnn.cpp chore(license): remove all license in file header 3 years ago
softmax.cpp chore(license): remove all license in file header 3 years ago
specializations.cpp fix(imperative): make functional ops support negative axis 3 years ago
tensor_manip.cpp chore(license): remove all license in file header 3 years ago
tensorrt_runtime.cpp chore(license): remove all license in file header 3 years ago
utility.cpp chore(license): remove all license in file header 3 years ago
vision.cpp feat(imperative): support empty tensor in roi_align 3 years ago
warp_affine.cpp chore(license): remove all license in file header 3 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