#0: update unpad doc for slice generality

tenstorrent · May 28, 2024 · 5ed51dc · 5ed51dc
1 parent 61aa564
commit 5ed51dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tt_eager/tt_lib/csrc/tt_lib_bindings_tensor_dm_ops.cpp b/tt_eager/tt_lib/csrc/tt_lib_bindings_tensor_dm_ops.cpp
@@ -300,8 +300,8 @@ namespace tt::tt_metal::detail{
                 :header: "Argument", "Description", "Data type", "Valid range", "Required"
 
                 "input", "Input tensor", "Tensor", "", "Yes"
-                "output_tensor_start", "Start indices of input tensor", "List[int[4]]", "Must be all 0s", "Yes"
-                "output_tensor_end", "End indices of input tensor in output tensor", "List[int[4]]", "Values along each dim must be < input_tensor_shape[i]", "Yes"
+                "output_tensor_start", "Start indices of input tensor", "List[int[tensor rank]]", "Values along each dim must be < input_tensor_shape[i]", "Yes"
+                "output_tensor_end", "End indices of input tensor in output tensor", "List[int[tensor rank]]", "Values along each dim must be < input_tensor_shape[i]", "Yes"
                 "pad_value", "Value to pad input tensor", "float", "", "Yes"
                 "output_mem_config", "Layout of tensor in TT Accelerator device memory banks", "MemoryConfig", "Default is interleaved in DRAM", "No"
         )doc");