| .. |
|
__init__.py
|
replace matmul/batchmatmul by mul when k is 1
|
2021-06-09 15:00:27 +08:00 |
|
_utils.py
|
fix static warnings
|
2021-06-04 17:25:42 +08:00 |
|
assign_add.py
|
add some expander ops
|
2021-03-18 19:03:36 +08:00 |
|
batchnorm.py
|
batchnorm expander supports when first input is float16
|
2021-04-27 09:46:42 +08:00 |
|
batchnorm_grad.py
|
[graph kernel] add expander ops.
|
2021-04-09 17:33:06 +08:00 |
|
bias_add.py
|
[graph kernel] add expander ops.
|
2021-04-09 17:33:06 +08:00 |
|
bias_add_grad.py
|
Refactor GraphKernelExpander (3rd submission)
|
2021-02-25 19:14:46 +08:00 |
|
clip_by_norm_no_div_sum.py
|
delete mark_interface_fusion and tensor reuse frontend pass for graph kernel
|
2021-03-27 11:45:04 +08:00 |
|
conv2d.py
|
expand conv2d when input format is DefaultFormat but attr format is NHWC
|
2021-05-28 11:38:32 +08:00 |
|
dropout_grad.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
expand_dims.py
|
[graph kernel] add expander ops.
|
2021-04-09 17:33:06 +08:00 |
|
fused_adam.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
fused_adam_weight_decay.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
fused_mul_add.py
|
adapt batchmatmul for GraphKernel in ascend
|
2021-05-10 10:13:56 +08:00 |
|
gelu.py
|
fix static warnings
|
2021-06-04 17:25:42 +08:00 |
|
gelu_grad.py
|
fix static warnings
|
2021-06-04 17:25:42 +08:00 |
|
gkdropout.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
lamb_apply_optimizer_assign.py
|
fix static warnings
|
2021-06-04 17:25:42 +08:00 |
|
lamb_apply_weight_assign.py
|
avoid overflow in realdiv
|
2021-06-07 11:46:07 +08:00 |
|
layernorm.py
|
adapt expanders of some ops from gpu to ascend
|
2021-05-04 16:32:45 +08:00 |
|
layernorm_grad.py
|
adapt for layernorm in ascend
|
2021-04-08 18:11:59 +08:00 |
|
logsoftmax.py
|
adapt for logsoftmax in ascend
|
2021-05-06 10:13:28 +08:00 |
|
logsoftmax_grad.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
matmul.py
|
replace matmul/batchmatmul by mul when k is 1
|
2021-06-09 15:00:27 +08:00 |
|
maximum_grad.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
minimum_grad.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
reduce_mean.py
|
Normalize the Reduce nodes' axis in GraphKernel
|
2021-03-16 15:55:50 +08:00 |
|
relu.py
|
add graph kernel expander ops.
|
2021-03-30 11:19:40 +08:00 |
|
relu_grad.py
|
add graph kernel expander ops.
|
2021-03-30 11:19:40 +08:00 |
|
sigmoid.py
|
[graph kernel] clean code for expanders.
|
2021-04-26 16:28:37 +08:00 |
|
sigmoid_cross_entropy_with_logits.py
|
rewrite sigmoid_cross_entropy_with_logits expander to avoid overflow
|
2021-05-11 14:37:48 +08:00 |
|
sigmoid_cross_entropy_with_logits_grad.py
|
[graph kernel] clean code for expanders.
|
2021-04-26 16:28:37 +08:00 |
|
sigmoid_grad.py
|
[graph kernel] clean code for expanders.
|
2021-04-26 16:28:37 +08:00 |
|
softmax.py
|
adapt expanders of some ops from gpu to ascend
|
2021-05-04 16:32:45 +08:00 |
|
softmax_cross_entropy_with_logits.py
|
fix expander bugs
|
2021-06-08 23:53:15 +08:00 |
|
softmax_grad_ext.py
|
expanders of some fusion ops
|
2021-05-06 10:23:41 +08:00 |
|
sqrt_grad.py
|
[graph kernel] clean code for expanders.
|
2021-04-26 16:28:37 +08:00 |
|
square.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
square_sum_v1.py
|
expanders of some fusion ops
|
2021-05-06 10:23:41 +08:00 |
|
squeeze.py
|
[GraphKernel] negative axis in Squeeze expander.
|
2021-04-25 17:31:04 +08:00 |
|
tanh_grad.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
tile.py
|
refactor tile op and in expander open on gpu
|
2021-04-25 17:42:37 +08:00 |
i-robot
07f58b0b46