| .. |
|
adam_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
adam_fusion.h
|
update adam_fusion and adam_weight_decay_fusion
|
2021-04-12 16:53:36 +08:00 |
|
adam_weight_decay_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
adam_weight_decay_fusion.h
|
update adam_fusion and adam_weight_decay_fusion
|
2021-04-12 16:53:36 +08:00 |
|
add_relu_grad_v2_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
add_relu_grad_v2_fusion.h
|
relu optimize
|
2020-12-09 14:31:45 +08:00 |
|
add_relu_v2_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
add_relu_v2_fusion.h
|
relu optimize
|
2020-12-09 14:31:45 +08:00 |
|
apply_momentum_scale_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
apply_momentum_scale_fusion.h
|
[auto-monad] Support side-effects by auto-monad
|
2021-02-08 09:01:15 +08:00 |
|
apply_momentum_weight_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
apply_momentum_weight_fusion.h
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
apply_momentum_weight_scale_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
apply_momentum_weight_scale_fusion.h
|
update apply_momentum_weight_scale_fusion pass
|
2021-05-11 19:12:07 +08:00 |
|
batch_norm_add_relu_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
batch_norm_add_relu_fusion.h
|
add bn fusion pass
|
2020-08-20 11:07:51 +08:00 |
|
batch_norm_add_relu_grad_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
batch_norm_add_relu_grad_fusion.h
|
gpu resnet50 fusion
|
2020-09-01 17:28:51 +08:00 |
|
batch_norm_relu_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
batch_norm_relu_fusion.h
|
add bn fusion pass
|
2020-08-20 11:07:51 +08:00 |
|
batch_norm_relu_grad_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
batch_norm_relu_grad_fusion.h
|
add bn fusion pass
|
2020-08-20 11:07:51 +08:00 |
|
bce_with_logits_loss_fusion.cc
|
add gpu BCEWithLogitsLoss kernel
|
2021-04-30 16:46:08 -04:00 |
|
bce_with_logits_loss_fusion.h
|
add gpu BCEWithLogitsLoss kernel
|
2021-04-30 16:46:08 -04:00 |
|
combine_cast_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
combine_cast_fusion.h
|
gpu add combine cast fusion
|
2020-09-27 16:07:07 +08:00 |
|
combine_momentum_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
combine_momentum_fusion.h
|
gpu add combine mom fusion
|
2020-09-27 10:09:06 +08:00 |
|
concat_outputs_for_all_gather.cc
|
[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset
|
2021-08-22 16:26:45 +08:00 |
|
concat_outputs_for_all_gather.h
|
Add all gather fusion and concat pass for gpu
|
2021-06-05 18:44:21 +08:00 |
|
cudnn_inplace_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
cudnn_inplace_fusion.h
|
cudnn inplace optimizer
|
2020-09-30 16:18:16 +08:00 |
|
insert_cast_gpu.cc
|
fix Conv3D precision under fp16
|
2021-08-25 13:06:44 +08:00 |
|
insert_cast_gpu.h
|
fix Conv3D precision under fp16
|
2021-08-25 13:06:44 +08:00 |
|
insert_format_transform_op.cc
|
[feat] [assistant] [I3T96T] add new Dataset operator CMUARCTICDataset
|
2021-08-22 16:26:45 +08:00 |
|
insert_format_transform_op.h
|
support int64 shape
|
2020-11-05 21:15:49 +08:00 |
|
matmul_biasadd_fusion.cc
|
gpu matmul biasadd fusion
|
2021-05-21 14:26:54 +08:00 |
|
matmul_biasadd_fusion.h
|
gpu matmul biasadd fusion
|
2021-05-21 14:26:54 +08:00 |
|
post_batch_norm_add_relu_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
post_batch_norm_add_relu_fusion.h
|
[auto-monad] Support side-effects by auto-monad
|
2021-02-08 09:01:15 +08:00 |
|
print_reduce_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
print_reduce_fusion.h
|
support string in GPU print
|
2021-03-12 14:12:19 -05:00 |
|
reduce_precision_fusion.cc
|
set "dst_type" attribute when creating Cast op
|
2021-08-26 14:19:54 +08:00 |
|
reduce_precision_fusion.h
|
gpu add reduce precision:int64->int32
|
2020-10-14 10:55:10 +08:00 |
|
relu_v2_pass.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
relu_v2_pass.h
|
relu optimize
|
2020-12-09 14:31:45 +08:00 |
|
remove_format_transform_pair.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
remove_format_transform_pair.h
|
gpu add format transform pass
|
2020-08-20 16:08:21 +08:00 |
|
remove_redundant_format_transform.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
remove_redundant_format_transform.h
|
gpu add the pass of remove redundant transpose
|
2020-09-02 14:16:21 +08:00 |
|
replace_addn_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
replace_addn_fusion.h
|
gpu add fusion: 1 replace bn cast 2 replace addn by tensoradd
|
2020-07-15 16:22:17 +08:00 |
|
replace_momentum_cast_fusion.cc
|
code review fix
|
2021-07-27 11:32:41 +08:00 |
|
replace_momentum_cast_fusion.h
|
gpu add fusion: replace momentum cast
|
2020-07-16 18:52:26 +08:00 |
2794b715e5