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.
 
 
 
 
 
 
He Wei 43e0967024 Decouple ir::Tensor class from python 5 years ago
..
addn_fission.cc Fix inputs size and attr for AddN fission pass 6 years ago
addn_fission.h implement AddN fission pass 6 years ago
batch_norm_bert_fission.cc add signle batchnorm fission pass 5 years ago
batch_norm_bert_fission.h Add batch norm bert fission pass 5 years ago
batch_norm_grad_infer_fission.cc Add tuple_getitem check for outputs of bn 5 years ago
batch_norm_grad_infer_fission.h Add batch_norm_grad infer fisson 5 years ago
batch_norm_grad_split.cc Support specifying reshape type for batchnorm fused op 5 years ago
batch_norm_grad_split.h Add BatchNormGrad split pass 5 years ago
bn_grad_split.cc initial version 6 years ago
bn_grad_split.h initial version 6 years ago
bn_split.cc fix exception when FusedBatchNorm's first input 's shape dims not equal 4 5 years ago
bn_split.h initial version 6 years ago
lars_v2_fission.cc add LarsV2 fission pass 5 years ago
lars_v2_fission.h add LarsV2 fission pass 5 years ago
layer_norm_grad_split.cc place layernormgrad split pass before kernel select 5 years ago
layer_norm_grad_split.h place layernormgrad split pass before kernel select 5 years ago
single_batch_norm_fission.cc fix the back outputs of BNTrainingUpdateV3 may cover the front whitch is empty 5 years ago
single_batch_norm_fission.h add signle batchnorm fission pass 5 years ago
split_fission.cc Add Split fission pass 5 years ago
split_fission.h Add Split fission pass 5 years ago
topk_split.cc Decouple ir::Tensor class from python 5 years ago
topk_split.h Check topk supported before converting input to attr 5 years ago
transdata_split.cc fix transdata insert cast 5 years ago
transdata_split.h move add_memcpy_async pass and transdata_split pass 5 years ago