Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk_schedule.cc38 depends_on_[&thunk].push_back(FindOrDie(hlo_to_thunk, &operand)); in AddDependenciesOnTransitiveOperands()
77 for (const auto& dependency : depends_on_) { in ThunkSchedule()
120 if (!depends_on_.contains(dst)) { in RemoveRedundantDependencyEdges()
126 std::list<const Thunk*>& sources = FindOrDie(depends_on_, dst); in RemoveRedundantDependencyEdges()
141 depends_on_.erase(dst); in RemoveRedundantDependencyEdges()
148 if (depends_on_.contains(thunk)) { in DependsOn()
149 return FindOrDie(depends_on_, thunk); in DependsOn()
182 for (const auto& entry : depends_on_) { in ToString()
Dthunk_schedule.h103 absl::flat_hash_map<const Thunk*, std::list<const Thunk*>> depends_on_; variable