16 Commits (1bc285a53a979ac07767701210c6afbaeadf711d)

Author	SHA1	Message	Date
Zeng Jinle	fff270eacd	follow comments,test=develop (#17273)	7 years ago
Kaipeng Deng	a71d8fdb87	Softmax_cross_entropy op add axis (#16806) ... * add attr axis infershape. test=develop * add CUDA kernel. test=develop * fix unittest. test=develop * fix unittest for soft_label. test=develop * fix fp16 unittest. test=develop * remove comment code. test=develop * refine test for axis. test=develop * add python api. test=develop * fix doc. test=develop * fix fp16 unittest. test=develop * fix ngraph test. test=develop * fix ENFORCE for test_imperative_transformer. test=develop * fit for ngraph test. test=develop * fix after rebase develop. test=develop * fix doc. test=develop * fix API.spec. test=develop * fix test_layers. test=develop * fix format. test=develop	7 years ago
Zeng Jinle	ee2028a110	Add use_cuda to inplace pass (#17205) ... * add use_cuda to inplace pass,test=develop * add test softmax_with_xe_inplace test,test=develop	7 years ago
Zeng Jinle	1202d3fc74	Refine model gpu memory (#16993) ... * speedup gc and inplace softmax_with_cross_entropy_grad test=develop * refine models gpu mem Merge skip vars and warning messages of mem opt remove relu mem opt test=develop * follow comments test=develop	7 years ago
phlrain	bbfc82cc42	softmax corss entropy support high rank ... test=develop	7 years ago
chengduo	efca4de78e	Fix cross_entropy bug (#16236) ... test=develop	7 years ago
Wu Yi	fd85418329	[Feature] support mix precision training for resnet (#14899) ... * clip softmax for fp16 * updates * fuse xent support fp16 test=develop * wip * wip * add simple row reduce * wip fp16 accurate softmax * add accurate softmax kernel for fp16 test=develop * update test=develop * fix cpu build test=develop * update api.spec test=develop * follow comments test=develop * fix build test=develop * fix trt build test=develop * fix inference build test=develop * fix merge test=develop * update test=develop * try fix build test=develop * fix build test=develop * rename real_exp test=develop * fortest * remove hacky kernels test=develop * clean up test=develop	7 years ago
Yu Yang	7604b1ad51	Fix Eigen macro when using GPU ... The macro should be defined by compiler rather than by source. test=develop	7 years ago
sneaxiy	5e5d2223a1	test=develop	7 years ago
Bai Yifan	e69d9c845b	code fix (#13365)	7 years ago
Bai Yifan	faf8ad2436	Add ignore_index in cross_entropy op (#13217) ... * add ignore index * update api.spec * enhance softmax_with_cross_entropy	7 years ago
sneaxiy	1b4515f6db	refine softmax_with_cross_entropy	7 years ago
qingqing01	b5a16dca20	Fix a critical bug in softmax_with_cross_entropy_op backward. (#9120) ... * Fix a critical bug in softmax_with_cross_entropy_op, which will lead to the wrong gradients. * Enhance unit testing.	8 years ago
qingqing01	24509f4af9	Fix the grammar in copyright. (#8403)	8 years ago
Yi Wang	fc374821dd	Correct #include path	8 years ago
Yi Wang	90648f336d	Move file to fluid/; Edit CMakeLists.txt	8 years ago