Huawei_Technology
/
MindSpore-AKG
Not watched
1
0
Fork 0
Code
Releases 1 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
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.
143 Commits
2 Branches
7.6 MB
378 Download
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'master'
${ noResults }
MindSpore-AKG/python/akg/ops/nn
xsmq 46f4c28fcf init 5 years ago
..
__init__.py init 5 years ago
avgpool.py init 5 years ago
batch_norm_ad.py init 5 years ago
batchmatmul.py init 5 years ago
bias_add.py init 5 years ago
bias_add_ad.py init 5 years ago
conv.py init 5 years ago
conv_backprop_filter.py init 5 years ago
conv_backprop_input.py init 5 years ago
conv_bn1.py init 5 years ago
conv_filter_ad.py init 5 years ago
conv_input_ad.py init 5 years ago
fused_batch_norm.py init 5 years ago
fused_batch_norm_grad.py init 5 years ago
fused_batch_norm_grad_split.py init 5 years ago
fused_batch_norm_split.py init 5 years ago
matmul.py init 5 years ago
maxpool.py init 5 years ago
maxpool_ad.py init 5 years ago
maxpool_grad.py init 5 years ago
maxpool_grad_with_argmax.py init 5 years ago
mean_ad.py init 5 years ago
relu.py init 5 years ago
relu_ad.py init 5 years ago
resize_nearest_neighbor_grad.py init 5 years ago
softmax.py init 5 years ago
sparse_softmax_cross_entropy_with_logits.py init 5 years ago
sparse_softmax_cross_entropy_with_logits_ad.py init 5 years ago
zeros_like.py init 5 years ago
Delete task

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

cancel
confirm

AKG(Auto Kernel Generator)对深度神经网络中的算子进行优化，并提供特定模式下的算子自动融合功能。AKG与MindSpore的图算融合功能协同工作，可提升在不同硬件后端上运行网络的性能。

Python C++ Text Cuda CMake other