-
Masaki Kozuki authored
* switch from clone to out-of-place subtract * Update apex/mpu/cross_entropy.py * Apply 1 suggestion(s) to 1 file(s) Co-authored-by: Eddie Yan <eddiey@nvidia.com>
* switch from clone to out-of-place subtract
* Update apex/mpu/cross_entropy.py
* Apply 1 suggestion(s) to 1 file(s)
Co-authored-by: Eddie Yan <eddiey@nvidia.com>