Searched refs:CopyNodes (Results 1 – 2 of 2) sorted by relevance
573 CopyNodes& copy_node = copy_map_[instruction]; in CreateCopyMap()635 const CopyNodes& copy_node = copy_map_.at(copy); in TryElideCopy()930 const CopyNodes& copy_info = pair.second; in ToString()952 struct CopyNodes { struct in xla::__anon49efd3e50111::CopyRemover957 absl::flat_hash_map<const HloInstruction*, CopyNodes> copy_map_;
126 void CopyNodes(Graph* graph, Zone* tmp_zone_, Node* dead, NodeRange nodes, in CopyNodes() function231 peeling.CopyNodes(graph_, tmp_zone_, dead, loop_tree_->BodyNodes(loop), in Peel()