Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_rematerialization.cc556 absl::Span<Item*> indirect_users);
1067 Item* original_item, Item* remat_item, absl::Span<Item*> indirect_users) { in AddRematerializedInstruction() argument
1098 const absl::flat_hash_set<Item*> indirect_users_set(indirect_users.begin(), in AddRematerializedInstruction()
1099 indirect_users.end()); in AddRematerializedInstruction()
1150 for (Item* indirect_user : indirect_users) { in AddRematerializedInstruction()
1562 absl::InlinedVector<Item*, 4> indirect_users; in RematerializeInstructions() local
1578 indirect_users.push_back(instruction_list->CreateItem(remat_use)); in RematerializeInstructions()
1588 indirect_users.push_back(instruction_list->CreateItem(remat_use)); in RematerializeInstructions()
1597 best_item, remat_item, absl::MakeSpan(indirect_users))); in RematerializeInstructions()
1605 const absl::flat_hash_set<Item*> indirect_users_set(indirect_users.begin(), in RematerializeInstructions()
[all …]