Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instruction.cc1851 if (operand->user_map_.find(this) != operand->user_map_.end()) { in DetachFromOperandsAndUsers()
2091 if (!ContainsKey(user_map_, user)) { in AddUser()
2092 user_map_.emplace(user, users_.size()); in AddUser()
2098 auto result = user_map_.find(user); in UserId()
2099 CHECK(result != user_map_.end()); in UserId()
2284 auto map_it = user_map_.find(user); in RemoveUser()
2285 CHECK(map_it != user_map_.end()); in RemoveUser()
2292 user_map_[users_.back()] = index; in RemoveUser()
2296 user_map_.erase(map_it); in RemoveUser()
2409 user_map_.clear(); in ReplaceAllUsesWithDifferentShape()
Dhlo_instruction.h1096 return ContainsKey(instruction->user_map_, this); in IsUserOf()
2083 absl::flat_hash_map<const HloInstruction*, int64> user_map_; variable