Searched defs:updates_flat (Results 1 – 3 of 3) sorted by relevance
139 auto updates_flat = in DoCompute() local227 auto updates_flat = in DoCompute() local
673 auto updates_flat = updates.shaped<T, 2>({num_updates, slice_size}); in DoScatterNd() local
805 auto updates_flat = updates.shaped<T, 2>({N, num_updates / N}); in Compute() local