| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 820 return ReplaceWithNewInstruction( in HandleAdd()
845 return ReplaceWithNewInstruction( in HandleAdd()
888 return ReplaceWithNewInstruction(add, std::move(dynamic_update_slice_full)); in HandleAdd()
918 return ReplaceWithNewInstruction( in HandleAdd()
1043 return ReplaceWithNewInstruction( in HandleAdd()
1104 TF_RETURN_IF_ERROR(ReplaceWithNewInstruction( in TrySimplifyTautologicalCompare()
1162 return ReplaceWithNewInstruction( in HandleBitcast()
1182 return ReplaceWithNewInstruction( in HandleCopy()
1200 return ReplaceWithNewInstruction( in HandleCopy()
1408 return ReplaceWithNewInstruction( in HandleConstant()
[all …]
|
| D | zero_sized_hlo_elimination.cc | 47 TF_RETURN_IF_ERROR(comp->ReplaceWithNewInstruction( in Run()
|
| D | transpose_folding.cc | 141 return dot->parent()->ReplaceWithNewInstruction(dot, std::move(new_dot)); in FoldTransposeIntoDot()
210 TF_CHECK_OK(convolution.parent()->ReplaceWithNewInstruction( in FoldTransposeIntoConvolution()
|
| D | hlo_constant_folding.cc | 150 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in Run()
|
| D | dynamic_index_splitter.cc | 91 TF_CHECK_OK(parent->ReplaceWithNewInstruction(dynamic_op, in Run()
|
| D | tree_reduction_rewriter.cc | 92 return ReplaceWithNewInstruction(hlo, std::move(new_output)); in HandleReduce()
|
| D | dfs_hlo_visitor_with_default.h | 295 Status ReplaceWithNewInstruction( in ReplaceWithNewInstruction() function
301 TF_RETURN_IF_ERROR(old_instruction->parent()->ReplaceWithNewInstruction( in ReplaceWithNewInstruction()
|
| D | convolution_group_converter.cc | 408 TF_CHECK_OK(computation_->ReplaceWithNewInstruction( in HandleBatchGroupCount()
484 return computation_->ReplaceWithNewInstruction( in HandleConvolution()
571 return computation_->ReplaceWithNewInstruction( in HandleConvolution()
|
| D | batchnorm_expander.cc | 289 TF_CHECK_OK(ReplaceWithNewInstruction(batch_norm, std::move(tuple))); in HandleBatchNormTraining()
554 TF_CHECK_OK(ReplaceWithNewInstruction(batch_norm, std::move(tuple))); in HandleBatchNormGrad()
|
| D | reshape_mover.cc | 255 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in PerformSinkReshapeOrTranspose()
|
| D | while_loop_simplifier.cc | 616 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in TryRemoveConstantParams()
941 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in TryFlattenNestedTuples()
1182 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in TryMergeInductionVariables()
|
| D | all_reduce_combiner.cc | 101 TF_RETURN_IF_ERROR(computation.ReplaceWithNewInstruction( in CombineAllReduces()
|
| D | hlo_computation.h | 335 Status ReplaceWithNewInstruction(
|
| D | hlo_computation.cc | 893 Status HloComputation::ReplaceWithNewInstruction( in ReplaceWithNewInstruction() function in xla::HloComputation
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | reduction_degenerate_dim_remover.cc | 71 return ReplaceWithNewInstruction(instr, std::move(reshape)); in HandleReduce()
87 return ReplaceWithNewInstruction(instr, std::move(new_reduce)); in HandleReduce()
|
| D | gemm_rewriter.cc | 72 ReplaceWithNewInstruction(instr, std::move(gemm_call))); in HandleDot()
120 ReplaceWithNewInstruction(instr, std::move(gemm_call))); in HandleAdd()
|
| D | cudnn_batchnorm_rewriter.cc | 115 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchNormInference()
204 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchNormTraining()
294 TF_RETURN_IF_ERROR(computation_->ReplaceWithNewInstruction( in HandleBatchNormGrad()
|
| D | tree_reduction_rewriter.cc | 196 return ReplaceWithNewInstruction(hlo, std::move(outer_reduce)); in RewriteReduction()
221 return ReplaceWithNewInstruction(hlo, std::move(out)); in RewriteBatchDimensionLargerThanTile()
|
| D | reduction_dimension_grouper.cc | 96 return ReplaceWithNewInstruction(reduce, std::move(new_reduce)); in HandleReduce()
|
| D | reduction_layout_normalizer.cc | 119 return ReplaceWithNewInstruction(reduce, std::move(new_reduce)); in HandleReduce()
|
| D | reduction_splitter.cc | 106 return ReplaceWithNewInstruction(reduce, std::move(final_reduce)); in HandleReduce()
|
| D | cudnn_fused_conv_rewriter.cc | 287 TF_RETURN_IF_ERROR(computation->ReplaceWithNewInstruction( in RunFuseBiasSideActivation()
|
| D | gpu_conv_rewriter.cc | 709 TF_RETURN_IF_ERROR(conv->parent()->ReplaceWithNewInstruction( in RunOnInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | conv_canonicalization.cc | 140 TF_RETURN_IF_ERROR(module->entry_computation()->ReplaceWithNewInstruction( in Run()
|