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.
 
 
 
 
 
 
lingyunli63 dd48f10c3d add assign ops in composite_topi 5 years ago
..
__init__.py add assign ops in composite_topi 5 years ago
abs.py GraphKernel supports GPU 5 years ago
add.py GraphKernel supports GPU 5 years ago
add_n.py GraphKernel supports GPU 5 years ago
assign.py add assign ops in composite_topi 5 years ago
cast.py gpu update cast type 5 years ago
equal.py equal like kernel support int8 and fp64 5 years ago
exp.py GraphKernel supports GPU 5 years ago
expand_dims.py GraphKernel supports GPU 5 years ago
greater.py fused select and greater op to improve bert perfermance on GPU 5 years ago
greater_equal.py equal like kernel support int8 and fp64 5 years ago
hsigmoid.py Refactor the akg op registers. 5 years ago
hsigmoid_grad.py Refactor the akg op registers. 5 years ago
hswish.py Refactor the akg op registers. 5 years ago
hswish_grad.py Refactor the akg op registers. 5 years ago
lessequal.py equal like kernel support int8 and fp64 5 years ago
log.py GraphKernel supports GPU 5 years ago
logical_and.py Refactor the akg op registers. 5 years ago
logical_not.py Refactor the akg op registers. 5 years ago
logical_or.py Refactor the akg op registers. 5 years ago
maximum.py GraphKernel supports GPU 5 years ago
mean.py Refactor the akg op registers. 5 years ago
mean_grad.py Refactor the akg op registers. 5 years ago
minimum.py GraphKernel supports GPU 5 years ago
mul.py Refactor the akg op registers. 5 years ago
neg.py GraphKernel supports GPU 5 years ago
notequal.py int16 uint8 bool supported akg ops 5 years ago
pow.py GraphKernel supports GPU 5 years ago
real_div.py GraphKernel supports GPU 5 years ago
reciprocal.py GraphKernel supports GPU 5 years ago
reduce_max.py GraphKernel supports GPU 5 years ago
reduce_min.py GraphKernel supports GPU 5 years ago
reduce_sum.py GraphKernel supports GPU 5 years ago
relu6.py Refactor the akg op registers. 5 years ago
relu6_grad.py Refactor the akg op registers. 5 years ago
reshape.py GraphKernel supports GPU 5 years ago
round.py GraphKernel supports GPU 5 years ago
rsqrt.py GraphKernel supports GPU 5 years ago
select.py fused select and greater op to improve bert perfermance on GPU 5 years ago
sqrt.py GraphKernel supports GPU 5 years ago
squeeze.py int16 uint8 bool supported akg ops 5 years ago
squeeze_grad.py int16 uint8 bool supported akg ops 5 years ago
sub.py Refactor the akg op registers. 5 years ago
tile.py int16 uint8 bool supported akg ops 5 years ago