Home
last modified time | relevance | path

Searched refs:ReplaceWithNewInstruction (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc366 Status ReplaceWithNewInstruction( in ReplaceWithNewInstruction() function in xla::__anon4eef97bd0111::AlgebraicSimplifierVisitor
372 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in ReplaceWithNewInstruction()
507 return ReplaceWithNewInstruction( in HandleAdd()
524 return ReplaceWithNewInstruction( in HandleAdd()
556 return ReplaceWithNewInstruction( in HandleAdd()
607 return ReplaceWithNewInstruction( in HandleBitcast()
628 return ReplaceWithNewInstruction( in HandleCopy()
812 return ReplaceWithNewInstruction( in HandleConstant()
821 return ReplaceWithNewInstruction( in HandleConstant()
841 return ReplaceWithNewInstruction( in HandleSubtract()
[all …]
Dbatchnorm_expander.cc139 Status ReplaceWithNewInstruction( in ReplaceWithNewInstruction() function in xla::__anon406107c40111::BatchNormExpanderVisitor
142 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in ReplaceWithNewInstruction()
322 TF_CHECK_OK(ReplaceWithNewInstruction(batch_norm, std::move(tuple))); in HandleBatchNormTraining()
427 ReplaceWithNewInstruction(batch_norm, std::move(shifted_normalized))); in HandleBatchNormInference()
605 TF_CHECK_OK(ReplaceWithNewInstruction(batch_norm, std::move(tuple))); in HandleBatchNormGrad()
Dconvolution_group_converter.cc311 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchGroupCount()
384 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleConvolution()
453 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleConvolution()
531 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleConvolution()
Dtranspose_folding.cc112 return dot->parent()->ReplaceWithNewInstruction(dot, std::move(new_dot)); in FoldTransposeIntoDot()
181 TF_CHECK_OK(convolution.parent()->ReplaceWithNewInstruction( in FoldTransposeIntoConvolution()
Dzero_sized_hlo_elimination.cc46 TF_RETURN_IF_ERROR(comp->ReplaceWithNewInstruction( in Run()
Dhlo_constant_folding.cc150 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in Run()
Ddynamic_index_splitter.cc91 TF_CHECK_OK(parent->ReplaceWithNewInstruction(dynamic_op, in Run()
Dwhile_loop_simplifier.cc435 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in TryRemoveConstantParams()
729 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in TryFlattenNestedTuples()
970 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in TryMergeInductionVariables()
Dreshape_mover.cc254 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in PerformSinkReshapeOrTranspose()
Dhlo_computation.h277 Status ReplaceWithNewInstruction(
Dhlo_computation.cc755 Status HloComputation::ReplaceWithNewInstruction( in ReplaceWithNewInstruction() function in xla::HloComputation
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_batchnorm_rewriter.cc85 computation_->ReplaceWithNewInstruction(batch_norm, std::move(libcall))); in HandleBatchNormInference()
153 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchNormTraining()
204 computation_->ReplaceWithNewInstruction(batch_norm, std::move(libcall))); in HandleBatchNormGrad()
Dcusolver_rewriter.cc150 TF_RETURN_IF_ERROR(instruction->parent()->ReplaceWithNewInstruction( in RunOnInstruction()
Dcudnn_fused_conv_rewriter.cc272 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in Run()
Dcudnn_conv_rewriter.cc546 TF_RETURN_IF_ERROR(conv->parent()->ReplaceWithNewInstruction( in RunOnInstruction()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc139 TF_RETURN_IF_ERROR(module->entry_computation()->ReplaceWithNewInstruction( in Run()