Searched refs:new_hlo (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_element_type_converter.cc | 184 HloInstruction* new_hlo; in Run() local
189 new_hlo = computation->AddInstruction( in Run()
191 TF_RETURN_IF_ERROR(new_hlo->CopyAllControlDepsFrom(hlo)); in Run()
193 new_hlo = ToElementType(new_hlo, eliminate_type_); in Run()
199 new_hlo = computation->AddInstruction( in Run()
201 TF_RETURN_IF_ERROR(new_hlo->CopyAllControlDepsFrom(hlo)); in Run()
205 new_hlo = ConvertTupleElements(new_hlo, old_shape); in Run()
207 new_hlo = computation->AddInstruction( in Run()
209 TF_RETURN_IF_ERROR(new_hlo->CopyAllControlDepsFrom(hlo)); in Run()
212 TF_RETURN_IF_ERROR(hlo->ReplaceAllUsesWith(new_hlo)); in Run()
|
| D | call_inliner.cc | 50 auto new_hlo = hlo->CloneWithNewOperands(hlo->shape(), new_operands); in DefaultAction() local
52 outer_->AddInstruction(std::move(new_hlo)); in DefaultAction()
108 HloInstruction* new_hlo) { in NoteMapping() argument
110 std::make_pair(subcomputation_hlo, new_hlo)); in NoteMapping()
|
| D | bfloat16_normalization.cc | 131 auto new_hlo, in ConvertType()
146 return new_hlo; in ConvertType()
155 TF_ASSIGN_OR_RETURN(auto new_hlo, ConvertType(hlo, from, to, computation)); in InsertConvertAfterOutput()
156 if (new_hlo == hlo) { in InsertConvertAfterOutput()
161 TF_RETURN_IF_ERROR(hlo->ReplaceUseWithDifferentShape(user, new_hlo)); in InsertConvertAfterOutput()
164 computation->set_root_instruction(new_hlo, /*accept_different_shape=*/true); in InsertConvertAfterOutput()
187 auto new_hlo, in ChangeOutputTypeThenInsertConvertBack()
208 TF_RETURN_IF_ERROR(hlo->ReplaceUseWithDifferentShape(user, new_hlo)); in ChangeOutputTypeThenInsertConvertBack()
212 computation->set_root_instruction(new_hlo, /*accept_different_shape=*/true); in ChangeOutputTypeThenInsertConvertBack()
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 472 HloInstruction* new_hlo = func(); in SetPartitionedHlo() local
473 new_hlo->set_sharding(hlo->sharding()); in SetPartitionedHlo()
474 new_hlo->set_metadata(hlo->metadata()); in SetPartitionedHlo()
476 hlo, PartitionedHlo(new_hlo, hlo->shape(), MakePartitioningState())); in SetPartitionedHlo()
|