#7296: Update ttnn docs

tenstorrent · Apr 18, 2024 · 1ba6add · 1ba6add
1 parent 9449087
commit 1ba6add
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ttnn/ttnn/operations/math.py b/ttnn/ttnn/operations/math.py
@@ -191,7 +191,7 @@ def _math_binary_validate_input_tensors(operation_name, input_tensor_a, input_te
     )
     def math_binary_function(
         input_tensor_a: ttnn.Tensor,
-        input_tensor_b: Union[ttnn.Tensor, int, float],
+        input_tensor_b: ttnn.Tensor,
         *,
         memory_config: ttnn.MemoryConfig = ttnn.DRAM_MEMORY_CONFIG,
     ) -> ttnn.Tensor:
@@ -220,7 +220,7 @@ def math_binary_function(
             Performs eltwise-binary {op_name} operation on two tensors :attr:`input_a` and :attr:`input_b`.
 
             .. math::
-                {name.replace('_',' ')}(\\mathrm{{input\\_tensor\\_a}}_i \\; , \\; \\mathrm{{input\\_tensor\\_b}}_i  \\; \\; or \\; \\; \\mathrm{{scalar}})
+                {name.replace('_',' ')}(\\mathrm{{input\\_tensor\\_a}}_i \\; , \\; \\mathrm{{input\\_tensor\\_b}}_i )
 
             Args:
                 * :attr:`input_tensor_a`