Skip to content

Commit 991c37f

Browse files
Arm backend: Remove stale broadcast helper docs (pytorch#19367)
Update the broadcast_tensors() docstring now that the old reshape_for_broadcast helper has been removed. Signed-off-by: Sebastian Larsson <sebastian.larsson@arm.com>
1 parent 2f566e3 commit 991c37f

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

backends/arm/tosa/utils.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,16 +64,14 @@ def are_fake_tensors_broadcastable(
6464
def broadcast_tensors(tosa_fb, nodes: list[Node]) -> list[Any]:
6565
"""Broadcast the FX nodes to a shared shape inside the TOSA graph.
6666
67-
This mirrors ``reshape_for_broadcast`` but also emits the tile operators
68-
needed to materialize the broadcast and supports any number of inputs.
67+
This emits the reshape and tile operators needed to materialize the
68+
broadcast and supports any number of inputs.
6969
7070
Args:
7171
tosa_fb (Any): TOSA graph builder that receives the broadcast
7272
operators.
7373
nodes (list[Node]): FX nodes whose tensor metadata should be
7474
broadcast.
75-
tosa_spec (TosaSpecification): Active TOSA specification used to
76-
decode tensor metadata.
7775
7876
Returns:
7977
list[Any]: Broadcast versions of the inputs. Each element is either

0 commit comments

Comments
 (0)