xla
a5d39687 - Properly lower select_scatter (#4793)

Commit
2 years ago
Properly lower select_scatter (#4793) * Properly lower select_scatter * Run linter * Update select_scatter to use kReshape and kNarrow * Update start_indices to UpdateSlice op * Remove unnecessary intermediate tensor in select_scatter op * Use original mutated_view tensor * Refactor select_scatter op
Author
Parents
Loading