@@ -95,8 +95,8 @@ void CrossEntropyWithSoftmaxGradCPUKernel(const CPUContext& dev_ctx,
9595 const int remain = d / axis_dim;
9696 for (int i = 0 ; i < n; ++i) { // for each sample_1_dim
9797 for (int j = 0 ; j < remain; j++) { // for each sample_other_dims
98- int idx = i * remain + j; // this sample's label_idx. for 1d case,
99- // remain=1 and j=0, so, idx = i
98+ auto idx = i * remain + j; // this sample's label_idx. for 1d case,
99+ // remain=1 and j=0, so, idx = i
100100 auto lbl = static_cast <int64_t >(label_data[idx]); // NOLINT
101101 if (lbl == ignore_index) {
102102 for (int k = 0 ; k < axis_dim; ++k) { // for each class id's label
@@ -147,8 +147,8 @@ void CrossEntropyWithSoftmaxGradCPUKernel(const CPUContext& dev_ctx,
147147 const int remain = d / axis_dim;
148148 for (int i = 0 ; i < n; ++i) { // for each sample_1_dim
149149 for (int j = 0 ; j < remain; j++) { // for each sample_other_dims
150- int idx = i * remain + j; // this sample's label_idx. for 1d case,
151- // remain=1 and j=0, so, idx = i
150+ auto idx = i * remain + j; // this sample's label_idx. for 1d case,
151+ // remain=1 and j=0, so, idx = i
152152 auto lbl = static_cast <int64_t >(label_data[idx]); // NOLINT
153153 if (lbl == ignore_index) {
154154 for (int k = 0 ; k < axis_dim; ++k) { // for each class id's label
0 commit comments