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.

JamesLim 45c7d90564 Optimization of elementwise CUDA kernel (#30801)		4 years ago
..
mkldnn	[oneDNN] elementwise add bf16 grad kernel with broadcasting (#31385)	4 years ago
CMakeLists.txt	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_add_op.cc	use iwyu clean include second time, test=develop (#30829)	4 years ago
elementwise_add_op.cu	Revert "improve elementwise_add_grad perf (#29277)" (#29464)	5 years ago
elementwise_add_op.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_add_op_xpu.cc	add aarch64 and sunway kunlun lib (#30027)	4 years ago
elementwise_div_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_div_op.cu	change the grad of div when complex types (#29804)	5 years ago
elementwise_div_op.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_div_op_xpu.cc	1. fix elementwise ops'bug 2. fix softmax_with_cross_entropy_op 3. add biliner_interp_op (#29448)	5 years ago
elementwise_floordiv_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_floordiv_op.cu	revert divide (#27202)	5 years ago
elementwise_floordiv_op.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_floordiv_op_xpu.cc	add xpu elementwise ops (#29031)	5 years ago
elementwise_max_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_max_op.cu	…
elementwise_max_op.h	…
elementwise_max_op_xpu.cc	1. fix elementwise ops'bug 2. fix softmax_with_cross_entropy_op 3. add biliner_interp_op (#29448)	5 years ago
elementwise_min_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_min_op.cu	…
elementwise_min_op.h	use iwyu clean include (#27267)	5 years ago
elementwise_min_op_xpu.cc	1. fix elementwise ops'bug 2. fix softmax_with_cross_entropy_op 3. add biliner_interp_op (#29448)	5 years ago
elementwise_mod_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_mod_op.cu	…
elementwise_mod_op.h	fix the remainder (#26995)	5 years ago
elementwise_mul_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_mul_op.cu	change grad elementwise_mul for complex types (#29757)	5 years ago
elementwise_mul_op.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_mul_op_xpu.cc	1. fix elementwise ops'bug 2. fix softmax_with_cross_entropy_op 3. add biliner_interp_op (#29448)	5 years ago
elementwise_op.h	[oneDNN] elementwise add bf16 grad kernel with broadcasting (#31385)	4 years ago
elementwise_op_function.cu.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_op_function.h	Optimization of elementwise CUDA kernel (#30801)	4 years ago
elementwise_pow_op.cc	add the op version check for the elementwise ops, test=op_version (#30010)	5 years ago
elementwise_pow_op.cu	…
elementwise_pow_op.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_pow_op_xpu.cc	add xpu elementwise ops (#29031)	5 years ago
elementwise_sub_op.cc	use iwyu clean include second time, test=develop (#30829)	4 years ago
elementwise_sub_op.cu	add complex64 and complex128 type; add +-*/@ and slice opreator for c… (#29199)	5 years ago
elementwise_sub_op.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
elementwise_sub_op_xpu.cc	add xpu elementwise ops (#29031)	5 years ago
elementwise_xpu.h	update kunlun conv2d/softmax/elementwise implemetation (#29229)	5 years ago
test_elementwise_add_grad_grad.cc	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
test_elementwise_add_op_inplace.cc	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
test_elementwise_div_grad_grad.cc	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
test_elementwise_op_grad_grad.h	[ROCM] update fluid elementwise op for rocm (part10), test=develop (#31361)	4 years ago
unity_build_rule.cmake	Compiling operator libraries with Unity build (#29130)	5 years ago