Searched refs:flat_updates (Results 1 – 4 of 4) sorted by relevance
117 auto flat_updates = updates_tensor.flat<int32>(); in FuzzImpl() local119 flat_updates(i) = 1; in FuzzImpl()
57 flat_updates = updates.reshape((num_updates, slice_size))62 flat_updates[ix_updates])
58 flat_updates = updates.reshape((num_updates, slice_size))63 flat_updates[ix_updates])
71 flat_updates = updates.reshape((num_updates, slice_size))76 flat_updates[ix_updates])