Skip to content

Commit 4ceaf80

Browse files
committed
Multigather code polish
1 parent 2f1f978 commit 4ceaf80

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

Multigather.md

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -64,11 +64,6 @@ function gatherMultiaxis(input, indices, axes)
6464
endfor
6565
output = new tensor(input.dataType, outputShape)
6666

67-
// Broadcast the indices for any non-gather dimensions like leading batches.
68-
// (note an efficient implementation would avoid the memory copy and just
69-
// use broadcasted zero strides)
70-
broadcastIndices = broadcast(indices, outputShapeTimesCoordinateSize)
71-
7267
for each outputCoordinate in output.coordinates
7368
// Determine the corresponding input coordinate to read from given the
7469
// current out coordinate and indices tensor coordinate. Use the dot

0 commit comments

Comments
 (0)