You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

jakpiase 5b4f8aac82 Added LSTM BF16 and fixed GRU BF16 (#31234)		4 years ago
..
mkldnn	Added LSTM BF16 and fixed GRU BF16 (#31234)	4 years ago
CMakeLists.txt	[ROCM] update fluid operators for rocm (part3), test=develop (#31213)	4 years ago
conv_fusion_op.cc	Op(conv2d_fusion) error message enhancement. (#23596)	5 years ago
conv_fusion_op.cu	Add tf32 switch for cuDNN (#29192)	4 years ago
fused_bn_activation_op.cc	Add bn and relu fuse pass (#22048)	5 years ago
fused_bn_activation_op.cu	Modify relu native implementation 2 (#30996)	4 years ago
fused_bn_activation_op.h	improve efficiency of runtime InferVarType (#22778)	5 years ago
fused_bn_add_activation_op.cc	add Fuse bn add act pass (#28196)	4 years ago
fused_bn_add_activation_op.cu	add Fuse bn add act pass (#28196)	4 years ago
fused_bn_add_activation_op.h	add Fuse bn add act pass (#28196)	4 years ago
fused_elemwise_activation_op.cc	register OPMaker and Infer Shape Check for fused_elementwise_add (#30259)	4 years ago
fused_elemwise_activation_op.cu	reduce the occupied size of memory for the fused pattern of elementwise_add Op and activation Op(relu Op for example) (#29885)	4 years ago
fused_elemwise_activation_op.h	refine fused_elemwise_activation error message (#27734)	4 years ago
fused_embedding_eltwise_layernorm_op.cc	Delete Ref & VectorRef and add GetDataSafely (#22997)	5 years ago
fused_embedding_eltwise_layernorm_op.cu	[ROCM] update fluid operators for rocm (part3), test=develop (#31213)	4 years ago
fused_embedding_fc_lstm_op.cc	fix error msg of fused_embedding_fc_lstm_op, test=develop (#27231)	4 years ago
fused_embedding_fc_lstm_op.h	…
fused_embedding_seq_pool_op.cc	Add macro BOOST_GET to enrich the error information of boost :: get (#24175)	5 years ago
fused_embedding_seq_pool_op.h	Polish the error message of fc, fused_fc_elementwise_layernorm and fused_embedding_seq_pool. (#27692)	4 years ago
fused_fc_elementwise_layernorm_op.cc	Polish some error message in opeators (#27876)	4 years ago
fused_fc_elementwise_layernorm_op.cu	[ROCM] update fluid operators for rocm (part3), test=develop (#31213)	4 years ago
fusion_conv_inception_op.cc	[error message enhancement] fused_elemwise_activation_op and fusion_conv_inception_op (#23686)	5 years ago
fusion_conv_inception_op.cu	Add tf32 switch for cuDNN (#29192)	4 years ago
fusion_group_op.cc	Add assign to fusion_group and enhance inplace execution in fusion_group. (#26121)	5 years ago
fusion_group_op.cu.cc	Add the support of fp16 in fusion_group (#22239)	5 years ago
fusion_group_op.h	Add assign to fusion_group and enhance inplace execution in fusion_group. (#26121)	5 years ago
fusion_group_op_test.cc	Fix gpu memory allocation bug. (#28703)	4 years ago
fusion_gru_op.cc	More precise mkldnn kernel rules in GetExpectedKernelType (#29840)	4 years ago
fusion_gru_op.h	…
fusion_lstm_op.cc	Added LSTM BF16 and fixed GRU BF16 (#31234)	4 years ago
fusion_lstm_op.h	…
fusion_repeated_fc_relu_op.cc	error message enhancement for py_func op. (#23565)	5 years ago
fusion_repeated_fc_relu_op.h	…
fusion_seqconv_eltadd_relu_op.cc	fix bug sequececonv_eltadd_relu_fuse_pass (#27404)	4 years ago
fusion_seqconv_eltadd_relu_op.h	…
fusion_seqexpand_concat_fc_op.cc	fusion_seqexpand_concat_fc error message enhancement, test=develop (#23558)	5 years ago
fusion_seqexpand_concat_fc_op.h	…
fusion_seqpool_concat_op.cc	error message enhancement for fusion_seqpool_concat_op. test=develop (#23563)	5 years ago
fusion_seqpool_concat_op.h	…
fusion_seqpool_cvm_concat_op.cc	enhance error msgs of fusion_seqpool_cvm_concat_op.cc, test=develop (#30240)	4 years ago
fusion_seqpool_cvm_concat_op.h	…
fusion_squared_mat_sub_op.cc	Op (FusionSquaredMatSub) error message enhancement. (#23498)	5 years ago
fusion_squared_mat_sub_op.h	…
fusion_transpose_flatten_concat_op.cc	Refine transpose flatten concat error message (#23625)	5 years ago
fusion_transpose_flatten_concat_op.cu.cc	use iwyu clean include (#27267)	4 years ago
fusion_transpose_flatten_concat_op.h	…
multi_gru_op.cc	Add multi_gru op and tests (#28591)	4 years ago
multi_gru_op.h	Add multi_gru op and tests (#28591)	4 years ago
multihead_matmul_op.cc	[BUG]: Head number can only be > 1 on multihead op (#23974)	5 years ago
multihead_matmul_op.cu	[ROCM] update fluid operators for rocm (part3), test=develop (#31213)	4 years ago
skip_layernorm_op.cc	裁剪transformer模型trt支持；修复tensorRT不支持DeletePass的bug (#28517)	4 years ago
skip_layernorm_op.cu	[ROCM] update fluid operators for rocm (part3), test=develop (#31213)	4 years ago
unity_build_rule.cmake	Compiling operator libraries with Unity build (#29130)	4 years ago