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/math_gpu
xsmq 46f4c28fcf init 5 years ago
..
__init__.py init 5 years ago
abs.py init 5 years ago
add.py init 5 years ago
addn.py init 5 years ago
batch_matmul.py init 5 years ago
cast.py init 5 years ago
divide.py init 5 years ago
equal.py init 5 years ago
exp.py init 5 years ago
expand_dims.py init 5 years ago
greater_equal.py init 5 years ago
less_equal.py init 5 years ago
log.py init 5 years ago
logical_and.py init 5 years ago
logical_not.py init 5 years ago
logical_or.py init 5 years ago
maximum.py init 5 years ago
minimum.py init 5 years ago
mul.py init 5 years ago
neg.py init 5 years ago
notequal.py init 5 years ago
pow.py init 5 years ago
reduce_max.py init 5 years ago
reduce_min.py init 5 years ago
reduce_sum.py init 5 years ago
round.py init 5 years ago
rsqrt.py init 5 years ago
select.py init 5 years ago
sqrt.py init 5 years ago
sub.py init 5 years ago
sum_value.py init 5 years ago
tensorcore_batch_matmul.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