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.
 
 
 
 
 
 
i-robot ab78bcf16f !19081 add ops IsNan 4 years ago
..
complex support complex in gpu about graph_kernel 4 years ago
__init__.py isnan isfinite isinf squaresumall identity oneslik 4 years ago
_utils.py add expander checkings 4 years ago
addn.py Add expander for AddN; update akg submodule 4 years ago
assign_add.py add some expander ops 5 years ago
batchnorm.py batchnorm expander supports when first input is float16 4 years ago
batchnorm_grad.py [graph kernel] add expander ops. 4 years ago
bias_add.py [graph kernel] add expander ops. 4 years ago
bias_add_grad.py Refactor GraphKernelExpander (3rd submission) 5 years ago
clip_by_norm_no_div_sum.py delete mark_interface_fusion and tensor reuse frontend pass for graph kernel 5 years ago
conv2d.py Enable TensorCore for Bert-Base on GPU, enable akg-conv2d for ssd on GPU 4 years ago
dropout_grad.py Refactor GraphKernelExpander (2nd submission) 5 years ago
equal_count.py !215 add ops: LogicalNot, LogicalAnd, LogicalOr, NotEqual, EqualCount, Asinh, Acosh 4 years ago
erfc.py add graph kernel div, floordiv, mod, floormod, floor 4 years ago
expand_dims.py [graph kernel] add expander ops. 4 years ago
fused_adam.py Refactor GraphKernelExpander (2nd submission) 5 years ago
fused_adam_weight_decay.py Refactor GraphKernelExpander (2nd submission) 5 years ago
fused_mul_add.py adapt batchmatmul for GraphKernel in ascend 4 years ago
gelu.py fix static warnings 4 years ago
gelu_grad.py fix static warnings 4 years ago
gkdropout.py Refactor GraphKernelExpander (2nd submission) 5 years ago
identity.py isnan isfinite isinf squaresumall identity oneslik 4 years ago
lamb_apply_optimizer_assign.py fix static warnings 4 years ago
lamb_apply_weight_assign.py avoid overflow in realdiv 4 years ago
layernorm.py adapt expanders of some ops from gpu to ascend 4 years ago
layernorm_grad.py adapt for layernorm in ascend 5 years ago
logsoftmax.py adapt for logsoftmax in ascend 4 years ago
logsoftmax_grad.py Refactor GraphKernelExpander (2nd submission) 5 years ago
matmul.py matmul to mul only work on aicore 4 years ago
maximum_grad.py fix maximum_grad and minimum_grad input_shape not equal to output_shape bug 4 years ago
minimum_grad.py fix maximum_grad and minimum_grad input_shape not equal to output_shape bug 4 years ago
oneslike.py isnan isfinite isinf squaresumall identity oneslik 4 years ago
reduce_mean.py Normalize the Reduce nodes' axis in GraphKernel 5 years ago
relu.py add graph kernel expander ops. 5 years ago
relu_grad.py add graph kernel expander ops. 5 years ago
sigmoid.py [graph kernel] clean code for expanders. 4 years ago
sigmoid_cross_entropy_with_logits.py rewrite sigmoid_cross_entropy_with_logits expander to avoid overflow 4 years ago
sigmoid_cross_entropy_with_logits_grad.py [graph kernel] clean code for expanders. 4 years ago
sigmoid_grad.py [graph kernel] clean code for expanders. 4 years ago
softmax.py adapt expanders of some ops from gpu to ascend 4 years ago
softmax_cross_entropy_with_logits.py fix expander bugs 4 years ago
softmax_grad_ext.py expanders of some fusion ops 4 years ago
sqrt_grad.py [graph kernel] clean code for expanders. 4 years ago
square.py Refactor GraphKernelExpander (2nd submission) 5 years ago
square_sum_all.py isnan isfinite isinf squaresumall identity oneslik 4 years ago
square_sum_v1.py add expander checkings 4 years ago
squared_difference.py [GraphKernel] add sponge ops. 4 years ago
squeeze.py [GraphKernel] negative axis in Squeeze expander. 4 years ago
tanh_grad.py Refactor GraphKernelExpander (2nd submission) 5 years ago
tile.py refactor tile op and in expander open on gpu 4 years ago