update class comments

Author: chilo-ms

onnxruntime/core/graph/data_propagation/add_op_data_propagation.h

@@ -18,14 +18,14 @@ namespace onnxruntime {
  *     |
  *     v
  *   Shape            (It saves [1, 3, 64, 64] in inferred_shape_values_ in output's node_arg
- *     |               after graph::SaveShapeValuesFromDataPropagation())
+ *     |               during Graph::SaveShapeValuesFromDataPropagation())
  *     |
  *     | ______
  *     |       |
  *     v       v
  *   Gather  Gather   (First 'Gather' saves 3 in inferred_scalar_value_ in output node_arg, and
  *     |       |       second 'Gather' saves 64 in inferred_scalar_value_ in output node_arg
- *     |       |       after graph::SaveShapeValuesFromDataPropagation(), if the 'index' attributes
+ *     |       |       during GatherOpDataPropagation(), if the 'index' attributes
  *     |       |       are 1 and 2 respectively)
  *      \     /
  *       \   /

onnxruntime/core/graph/data_propagation/custom_data_propagation.h

@@ -14,7 +14,8 @@ namespace onnxruntime {
  * @class CustomDataPropagation
  * Custom data propagation for the operator to help enhance shape inference.
  *
- * Calling infer() can infer the output values for the specific operator given the input is shape values.
+ * Calling infer() can infer the output values for the specific operator given the input is shape values
+ * and saves the output values in output node_arg for other operators to use later.
  * The purpose of this class is to make shape values being correctly inferred and propogated through the graph.
  */
 class CustomDataPropagationBase {

onnxruntime/core/graph/data_propagation/div_op_data_propagation.h

@@ -18,14 +18,14 @@ namespace onnxruntime {
  *     |
  *     v
  *   Shape            (It saves [1, 3, 64, 64] in inferred_shape_values_ in output's node_arg
- *     |               after graph::SaveShapeValuesFromDataPropagation())
+ *     |               during graph::SaveShapeValuesFromDataPropagation())
  *     |
  *     | ______
  *     |       |
  *     v       v
  *   Gather  Gather   (First 'Gather' saves 64 in inferred_scalar_value_ in output node_arg, and
  *     |       |       second 'Gather' saves 1 in inferred_scalar_value_ in output node_arg
- *     |       |       after graph::SaveShapeValuesFromDataPropagation(), if the 'index' attributes
+ *     |       |       during GatherOpDataPropagation(), if the 'index' attributes
  *     |       |       are 2 and 0 respectively)
  *      \     /
  *       \   /

onnxruntime/core/graph/data_propagation/mul_op_data_propagation.h

@@ -17,14 +17,14 @@ namespace onnxruntime {
  *     |
  *     v
  *   Shape            (It saves [1, 3, 64, 64] in inferred_shape_values_ in output's node_arg
- *     |               after graph::SaveShapeValuesFromDataPropagation())
+ *     |               during graph::SaveShapeValuesFromDataPropagation())
  *     |
  *     | ______
  *     |       |
  *     v       v
  *   Gather  Gather   (First 'Gather' saves 3 in inferred_scalar_value_ in output node_arg, and
  *     |       |       second 'Gather' saves 64 in inferred_scalar_value_ in output node_arg
- *     |       |       after graph::SaveShapeValuesFromDataPropagation(), if the 'index' attributes
+ *     |       |       during GatherOpDataPropagation(), if the 'index' attributes
  *     |       |       are 1 and 2 respectively)
  *      \     /
  *       \   /

onnxruntime/core/graph/data_propagation/size_op_data_propagation.h

@@ -8,6 +8,10 @@
 
 namespace onnxruntime {
 
+/**
+ * @brief Class to infer the output scalar for 'Size' operator given the input is shape values.
+ * 
+ */
 class SizeOpDataPropagation : public CustomDataPropagationBase {
  public:
   SizeOpDataPropagation(const Node& node,

onnxruntime/core/graph/data_propagation/sub_op_data_propagation.h

@@ -18,14 +18,14 @@ namespace onnxruntime {
  *     |
  *     v
  *   Shape            (It saves [1, 3, 64, 64] in inferred_shape_values_ in output's node_arg
- *     |               after graph::SaveShapeValuesFromDataPropagation())
+ *     |               during graph::SaveShapeValuesFromDataPropagation())
  *     |
  *     | ______
  *     |       |
  *     v       v
  *   Gather  Gather   (First 'Gather' saves 64 in inferred_scalar_value_ in output node_arg, and
  *     |       |       second 'Gather' saves 3 in inferred_scalar_value_ in output node_arg
- *     |       |       after graph::SaveShapeValuesFromDataPropagation(), if the 'index' attributes
+ *     |       |       during GatherOpDataPropagation(), if the 'index' attributes
  *     |       |       are 2 and 1 respectively)
  *      \     /
  *       \   /