diff --git a/orttraining/orttraining/training_ops/cuda/loss/softmax_cross_entropy_loss_impl.cc b/orttraining/orttraining/training_ops/cuda/loss/softmax_cross_entropy_loss_impl.cc index b10836d7a0..f2da682b67 100644 --- a/orttraining/orttraining/training_ops/cuda/loss/softmax_cross_entropy_loss_impl.cc +++ b/orttraining/orttraining/training_ops/cuda/loss/softmax_cross_entropy_loss_impl.cc @@ -4,8 +4,6 @@ #include "core/providers/cuda/math/softmax.h" #include "core/providers/cuda/reduction/reduction_functions.h" #include "core/providers/cuda/tensor/transpose.h" -#include "core/providers/cuda/math/unary_elementwise_ops_impl.h" -#include "core/providers/cpu/controlflow/scan_utils.h" #include "core/framework/ml_value.h" #include "orttraining/training_ops/cpu/loss/softmax_cross_entropy_loss.h" #include "orttraining/training_ops/cuda/loss/softmax_cross_entropy_loss_impl.h"