Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dinstruction_fusion.cc333 InsertOrDie(&post_order_index_, post_order_[i], i); in ReversePostOrderFusionQueue()
356 post_order_index_.erase(instruction); in DequeueNextInstructionAndOperandsToFuseInOrder()
406 if (!ContainsKey(post_order_index_, instruction->mutable_operand(i))) { in DequeueNextInstructionAndOperandsToFuseInOrder()
415 FindOrDie(post_order_index_, instruction->mutable_operand(i)) > in DequeueNextInstructionAndOperandsToFuseInOrder()
416 FindOrDie(post_order_index_, instruction->mutable_operand(j))); in DequeueNextInstructionAndOperandsToFuseInOrder()
427 InsertOrDie(&post_order_index_, fusion, post_order_.size()); in OnFusingInstruction()
432 post_order_[FindOrDie(post_order_index_, instruction)] = nullptr; in RemoveInstruction()
433 post_order_index_.erase(instruction); in RemoveInstruction()
438 absl::flat_hash_map<HloInstruction*, int> post_order_index_; member in xla::__anon67f11ea50611::ReversePostOrderFusionQueue