#3645: fix moreh_softmax, moreh_softmin comment

tenstorrent · Jan 25, 2024 · a7545af · a7545af
1 parent 2945865
commit a7545af
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tt_eager/tt_lib/csrc/operations/primary/module.hpp b/tt_eager/tt_lib/csrc/operations/primary/module.hpp
@@ -489,7 +489,7 @@ void py_module(py::module& m_primary) {
         py::arg("dim").noconvert(),
         py::arg("strategy").noconvert() = MorehSoftmaxOpParallelizationStrategy::NONE,
         py::arg("output_mem_config").noconvert() = operation::DEFAULT_OUTPUT_MEMORY_CONFIG,
-        "Performs a softmax operation. Returns a output tensor.");
+        "Performs a softmax operation. Returns an output tensor.");
     m_primary.def(
         "moreh_softmax_backward",
         &moreh_softmax_backward,
@@ -498,15 +498,15 @@ void py_module(py::module& m_primary) {
         py::arg("dim").noconvert(),
         py::arg("strategy").noconvert() = MorehSoftmaxBackwardOpParallelizationStrategy::NONE,
         py::arg("output_mem_config").noconvert() = operation::DEFAULT_OUTPUT_MEMORY_CONFIG,
-        "Performs a softmax backward operation. Returns a input grad tensor.");
+        "Performs a softmax backward operation. Returns an input grad tensor.");
     m_primary.def(
         "moreh_softmin",
         &moreh_softmin,
         py::arg("input_tensors").noconvert(),
         py::arg("dim").noconvert(),
         py::arg("strategy").noconvert() = MorehSoftmaxOpParallelizationStrategy::NONE,
         py::arg("output_mem_config").noconvert() = operation::DEFAULT_OUTPUT_MEMORY_CONFIG,
-        "Performs a softmax operation. Returns a output tensor.");
+        "Performs a softmin operation. Returns an output tensor.");
     m_primary.def(
         "moreh_softmin_backward",
         &moreh_softmin_backward,
@@ -515,7 +515,7 @@ void py_module(py::module& m_primary) {
         py::arg("dim").noconvert(),
         py::arg("strategy").noconvert() = MorehSoftmaxBackwardOpParallelizationStrategy::NONE,
         py::arg("output_mem_config").noconvert() = operation::DEFAULT_OUTPUT_MEMORY_CONFIG,
-        "Performs a softmin backward operation. Returns a input grad tensor.");
+        "Performs a softmin backward operation. Returns an input grad tensor.");
     m_primary.def(
         "moreh_sum",
         &moreh_sum,