Searched refs:copy_map_ (Results 1 – 1 of 1) sorted by relevance
591 CopyNodes& copy_node = copy_map_[instruction]; in CreateCopyMap()622 if (def->opcode() == HloOpcode::kCopy && ContainsKey(copy_map_, def)) { in Verify()623 TF_RET_CHECK(copy_map_.at(def).dest == p); in Verify()627 ContainsKey(copy_map_, use->instruction)) { in Verify()628 TF_RET_CHECK(copy_map_.at(use->instruction).src == p); in Verify()645 if (!ContainsKey(copy_map_, copy)) { in TryElideCopy()653 const CopyNodes& copy_node = copy_map_.at(copy); in TryElideCopy()843 ContainsKey(copy_map_, copy_use->instruction)) { in RemoveCopyValue()844 copy_map_.at(copy_use->instruction).src = operand_node; in RemoveCopyValue()849 copy_map_.erase(copy_value_node->value->defining_instruction()); in RemoveCopyValue()[all …]