Fix tile_dot docstring referencing nonexistent wp.tensordot

* Fix tile_dot docstring referencing nonexistent wp.tensordot

The "Equivalent to" line referred to wp.tensordot, which is not a
Warp builtin. Rewrite the docstring to describe the contraction by
element type (scalar, vector, matrix) and provide Python-level
equivalents only where they actually exist.

Signed-off-by: Eric Shi <ershi@nvidia.com>

Approved-by: Eric Shi <ershi@nvidia.com>

See merge request omniverse/warp!2311

Author: shi-eric

warp/__init__.pyi

@@ -3596,13 +3596,18 @@ def tile_sum(a: Tile[Any, tuple[int, ...]]) -> Tile[Any, tuple[Literal[1]]]:
 def tile_dot(a: Tile[Any, tuple[int, ...]], b: Tile[Any, tuple[int, ...]]) -> Tile[Any, tuple[Literal[1]]]:
     """Compute the dot product of two tiles.
 
-    Computes a full contraction (tensordot) between corresponding elements
-    and sums the results. For scalar tiles this is the standard dot product;
-    for vector or matrix tiles each element pair is fully contracted
-    (e.g., ``wp.dot(a[i], b[i])`` for ``vec3`` elements).
-
-    Equivalent to ``wp.tile_sum(wp.tile_map(wp.tensordot, a, b))``
-    but without any intermediate tiles or shared-memory round trips.
+    Computes a full contraction between corresponding elements and sums
+    the results. For scalar tiles this is the standard dot product; for
+    vector tiles each pair is contracted via ``wp.dot``; for matrix tiles
+    it is the Frobenius inner product (the sum of element-wise products
+    over all axes).
+
+    Equivalent in Python to ``wp.tile_sum(a * b)`` for scalar tiles and to
+    ``wp.tile_sum(wp.tile_map(wp.dot, a, b))`` for vector tiles, but
+    without the intermediate tile and shared-memory round trip the
+    explicit forms would require. Matrix-typed tiles have no
+    purely-elementwise Python equivalent because ``wp.dot`` is not
+    defined for matrices.
 
     Args:
         a: First tile operand.

warp/_src/builtins.py

@@ -5337,13 +5337,18 @@ def tile_dot_value_func(arg_types, arg_values):
     value_func=tile_dot_value_func,
     doc="""Compute the dot product of two tiles.
 
-    Computes a full contraction (tensordot) between corresponding elements
-    and sums the results. For scalar tiles this is the standard dot product;
-    for vector or matrix tiles each element pair is fully contracted
-    (e.g., ``wp.dot(a[i], b[i])`` for ``vec3`` elements).
-
-    Equivalent to ``wp.tile_sum(wp.tile_map(wp.tensordot, a, b))``
-    but without any intermediate tiles or shared-memory round trips.
+    Computes a full contraction between corresponding elements and sums
+    the results. For scalar tiles this is the standard dot product; for
+    vector tiles each pair is contracted via ``wp.dot``; for matrix tiles
+    it is the Frobenius inner product (the sum of element-wise products
+    over all axes).
+
+    Equivalent in Python to ``wp.tile_sum(a * b)`` for scalar tiles and to
+    ``wp.tile_sum(wp.tile_map(wp.dot, a, b))`` for vector tiles, but
+    without the intermediate tile and shared-memory round trip the
+    explicit forms would require. Matrix-typed tiles have no
+    purely-elementwise Python equivalent because ``wp.dot`` is not
+    defined for matrices.
 
     Args:
         a: First tile operand.