| .. |
|
complex
|
support complex in gpu about graph_kernel
|
2021-07-09 08:51:52 +08:00 |
|
__init__.py
|
support tot ops
|
2021-08-24 15:27:12 +08:00 |
|
_utils.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
addn.py
|
Add expander for AddN; update akg submodule
|
2021-06-29 11:43:46 +08:00 |
|
assign_add.py
|
add some expander ops
|
2021-03-18 19:03:36 +08:00 |
|
batchnorm.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
batchnorm_grad.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +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
|
Enable TensorCore for Bert-Base on GPU, enable akg-conv2d for ssd on GPU
|
2021-06-28 00:21:50 +08:00 |
|
dropout_grad.py
|
Refactor GraphKernelExpander (2nd submission)
|
2021-02-22 14:49:21 +08:00 |
|
equal_count.py
|
!215 add ops: LogicalNot, LogicalAnd, LogicalOr, NotEqual, EqualCount, Asinh, Acosh
|
2021-07-21 10:01:17 +08:00 |
|
erfc.py
|
add graph kernel div, floordiv, mod, floormod, floor
|
2021-07-22 14:16:56 +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
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
gather.py
|
support tot ops
|
2021-08-24 15:27:12 +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 |
|
identity.py
|
isnan isfinite isinf squaresumall identity oneslik
|
2021-07-26 12:58:30 +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
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +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
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
maximum_grad.py
|
fix maximum_grad and minimum_grad input_shape not equal to output_shape bug
|
2021-06-16 14:25:18 +08:00 |
|
minimum_grad.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
oneslike.py
|
isnan isfinite isinf squaresumall identity oneslik
|
2021-07-26 12:58:30 +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 |
|
slice.py
|
[GraphKernel] fix bert and add graph kernel ops.
|
2021-08-10 16:02:00 +08:00 |
|
softmax.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
softmax_cross_entropy_with_logits.py
|
fix expander bugs
|
2021-06-08 23:53:15 +08:00 |
|
softmax_grad_ext.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +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_all.py
|
isnan isfinite isinf squaresumall identity oneslik
|
2021-07-26 12:58:30 +08:00 |
|
square_sum_v1.py
|
Revert "[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset"
|
2021-08-23 01:46:38 +08:00 |
|
squared_difference.py
|
[GraphKernel] add sponge ops.
|
2021-07-19 19:34:23 +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 |
tronzhang
134ffa9efe