Searched refs:new_concat (Results 1 – 7 of 7) sorted by relevance
68 auto new_concat = comp->AddInstruction(concat->CloneWithNewOperands( in SplitConcatenate() local70 new_operands.push_back(new_concat); in SplitConcatenate()
121 auto new_concat = in handleCat() local123 new_concat->output()->setType(node->output()->type()); in handleCat()124 concats_to_replace_[node] = new_concat; in handleCat()166 auto new_concat = in handleCat() local168 new_concat->output()->setType(node->output()->type()); in handleCat()169 concats_to_replace_[node] = new_concat; in handleCat()
46 auto new_concat = parent->AddInstruction(HloInstruction::CreateConcatenate( in ConcatForwarding() local48 TF_RETURN_IF_ERROR(parent->ReplaceInstruction(concat, new_concat)); in ConcatForwarding()
1563 TF_ASSIGN_OR_RETURN(HloInstruction * new_concat, in TrySimplifyTautologicalBitcastConvert()1565 TF_RETURN_IF_ERROR(ReplaceInstruction(bitcast, new_concat)); in TrySimplifyTautologicalBitcastConvert()4743 HloInstruction * new_concat, in HandleSlice()4748 *new_concat->mutable_shape()->mutable_layout() = in HandleSlice()4750 simplifier_->UpdateLayout(new_concat->mutable_shape()); in HandleSlice()4751 concat->SetupDerivedInstruction(new_concat); in HandleSlice()4752 operand = new_concat; in HandleSlice()
2750 TF_ASSIGN_OR_RETURN(HloInstruction * new_concat, in PropagateOnConcat()2752 old_to_new_instrs_[concat] = new_concat; in PropagateOnConcat()2756 instr_to_dim_permute_map_[new_concat] = in PropagateOnConcat()
583 auto new_concat = rewriter.create<TFL::ConcatenationOp>( in matchAndRewrite() local588 auto output_reshape_op = InsertReshapeOp(concat_op.getLoc(), new_concat, in matchAndRewrite()
916 auto new_concat = builder.create<TFL::ConcatenationOp>( in fold() local920 return new_concat.getResult(); in fold()