Searched refs:flat_updates (Results 1 – 3 of 3) 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])
70 flat_updates = updates.reshape((num_updates, slice_size))75 flat_updates[ix_updates])