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.
 
 
 
 
 
 
zengzitao 62458b5636 adapt graph kernel for cpu 4 years ago
..
complex support complex in gpu about graph_kernel 4 years ago
__init__.py adapt graph kernel for cpu 4 years ago
_utils.py Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 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 Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
batchnorm_grad.py Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 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 fix master warnings 4 years ago
erfc.py add graph kernel div, floordiv, mod, floormod, floor 4 years ago
expand_dims.py [graph kernel] add expander ops. 5 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 Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
gather.py support tot ops 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 fix master warnings 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 Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 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 Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 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 Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
oneslike.py fix master warnings 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
slice.py [GraphKernel] fix bert and add graph kernel ops. 4 years ago
softmax.py Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 4 years ago
softmax_cross_entropy_with_logits.py fix expander bugs 4 years ago
softmax_grad_ext.py Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 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 fix master warnings 4 years ago
square_sum_v1.py Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset" 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