Home
last modified time | relevance | path

Searched refs:used_tuple_indices (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dwhile_loop_simplifier.cc44 HloInstruction* while_op, absl::flat_hash_set<int64>& used_tuple_indices) { in RemoveDeadTupleIndices() argument
46 std::vector<int64> new_to_old_tuple_idx(used_tuple_indices.begin(), in RemoveDeadTupleIndices()
47 used_tuple_indices.end()); in RemoveDeadTupleIndices()
264 absl::flat_hash_set<int64> used_tuple_indices; in TryRemoveDeadWhileParams() local
284 used_tuple_indices.insert(user->tuple_index()); in TryRemoveDeadWhileParams()
285 if (used_tuple_indices.size() == tuple_size) { in TryRemoveDeadWhileParams()
297 if (used_tuple_indices.contains(i)) { in TryRemoveDeadWhileParams()
307 used_tuple_indices.insert(i); in TryRemoveDeadWhileParams()
309 if (used_tuple_indices.size() == tuple_size) { in TryRemoveDeadWhileParams()
319 CHECK_LT(used_tuple_indices.size(), tuple_size); in TryRemoveDeadWhileParams()
[all …]