[cker] Fix wrong comment (#13985)

This commit fixes a wrong comment. - sparse shape -> broadcast shape ONE-DCO-1.0-Signed-off-by: ragmani <[email protected]>
Samsung · Sep 11, 2024 · 54b9838 · 54b9838
1 parent d8fa131
commit 54b9838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/compute/cker/include/cker/train/operation/Loss.h b/compute/cker/include/cker/train/operation/Loss.h
@@ -163,7 +163,7 @@ void CategoricalCrossEntropyWithLogits(const Shape &logits_shape, const T *logit
                                        const Shape &grad_shape, T *grad_data,
                                        LossReductionType reduction_type)
 {
-  // TODO Enable sparse shapes
+  // TODO Enable broadcast shapes
   if (loss_out_shape.DimensionsCount() != 1)
     throw std::runtime_error(
       "cker::CategoricalCrossEntropyWithLogits: loss output dimension count should be 1");