Searched defs:new_reduce (Results 1 – 5 of 5) sorted by relevance
77 std::unique_ptr<HloInstruction> new_reduce = HloInstruction::CreateReduce( in HandleReduce() local
92 std::unique_ptr<HloInstruction> new_reduce = HloInstruction::CreateReduce( in HandleReduce() local
108 std::unique_ptr<HloInstruction> new_reduce = HloInstruction::CreateReduce( in HandleReduce() local
255 auto new_reduce = rewriter.cloneWithoutRegions(reduce); in matchAndRewrite() local
4777 auto new_reduce = computation_->AddInstruction( in HandleReduce() local4835 HloInstruction* new_reduce = in HandleReduce() local4951 HloInstruction* new_reduce = computation_->AddInstruction( in HandleReduce() local5091 auto new_reduce = in HandleReduceWindow() local