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.

Leo Chen f354e1d6d5 [NPU] fix some op bugs (#31855) ... * fix some op bugs * fix some bugs * follow comments * fix log level * add ut		4 years ago
..
CMakeLists.txt	[NPU] Support npu kernel for amp_check_finite_and_unscale_npu op (#31457)	4 years ago
check_finite_and_unscale_op.cc	Update amp_check_finite_and_scale_op and add an updating_loss_scaling op for static graph amp training. (#26240)	5 years ago
check_finite_and_unscale_op.cu	Support pure fp16 training for AMP API. (#29544)	4 years ago
check_finite_and_unscale_op.h	Update amp_check_finite_and_scale_op and add an updating_loss_scaling op for static graph amp training. (#26240)	5 years ago
check_finite_and_unscale_op_npu.cc	[NPU] Support npu kernel for amp_check_finite_and_unscale_npu op (#31457)	4 years ago
check_finite_and_unscale_op_npu_test.cc	[NPU] Support npu kernel for amp_check_finite_and_unscale_npu op (#31457)	4 years ago
fp16_type_traits.h	Support pure fp16 training for AMP API. (#29544)	4 years ago
unity_build_rule.cmake	Compiling operator libraries with Unity build (#29130)	5 years ago
update_loss_scaling_op.cc	Support pure fp16 training for AMP API. (#29544)	4 years ago
update_loss_scaling_op.cu	Support pure fp16 training for AMP API. (#29544)	4 years ago
update_loss_scaling_op.h	Support pure fp16 training for AMP API. (#29544)	4 years ago
update_loss_scaling_op_npu.cc	[NPU] fix some op bugs (#31855)	4 years ago