Home
last modified time | relevance | path

Searched refs:RemoveInstruction (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_domain_remover.cc83 TF_RETURN_IF_ERROR(computation->RemoveInstruction(operand)); in Run()
92 TF_RETURN_IF_ERROR(computation->RemoveInstruction(root)); in Run()
Dhlo_cse.cc85 TF_CHECK_OK(computation->RemoveInstruction(instruction)); in CombineConstants()
159 TF_RETURN_IF_ERROR(computation->RemoveInstruction(instruction)); in Run()
Dhlo_computation.cc153 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveParameter()
162 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveParameter()
176 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveUnusedParameters()
188 TF_RETURN_IF_ERROR(RemoveInstruction(param_instruction)); in RemoveUnusedParameters()
244 TF_RETURN_IF_ERROR(RemoveInstruction(item)); in RemoveInstructionAndUnusedOperands()
250 Status HloComputation::RemoveInstruction(HloInstruction* instruction) { in RemoveInstruction() function in xla::HloComputation
604 TF_CHECK_OK(RemoveInstruction(root)); in FuseInstructionsInto()
609 TF_CHECK_OK(RemoveInstruction(instruction)); in FuseInstructionsInto()
Dfusion_queue.h48 virtual void RemoveInstruction(HloInstruction* instruction) = 0;
Dinstruction_fusion.cc431 void RemoveInstruction(HloInstruction* instruction) override { in RemoveInstruction() function in xla::__anon67f11ea50611::ReversePostOrderFusionQueue
516 fusion_queue->RemoveInstruction(operand); in Run()
518 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(operand)); in Run()
Ddefuser.cc86 fusion_instruction->parent()->RemoveInstruction(fusion_instruction)); in Defuse()
Dmap_inliner.cc84 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(map)); in HandleMap()
Dreduce_precision_insertion_test.cc412 EXPECT_IS_OK(computation->RemoveInstruction(y)); in TEST_F()
448 EXPECT_IS_OK(computation->RemoveInstruction(y)); in TEST_F()
495 EXPECT_IS_OK(computation->RemoveInstruction(y)); in TEST_F()
Dhlo_element_type_converter.cc218 TF_RETURN_IF_ERROR(computation->RemoveInstruction(hlo)); in Run()
Dhlo_domain_test.cc462 TF_EXPECT_OK(module->entry_computation()->RemoveInstruction(gte)); in TEST_F()
463 TF_EXPECT_OK(module->entry_computation()->RemoveInstruction(tuple)); in TEST_F()
Dhlo_computation.h136 Status RemoveInstruction(HloInstruction* instruction);
Dwhile_util.cc130 TF_RETURN_IF_ERROR(containing_computation->RemoveInstruction(while_instr)); in MakeInstructionsLiveIn()
Dar_crs_combiner.cc330 TF_CHECK_OK(parent_computation->RemoveInstruction(all_reduce)); in RewriteGraph()
Dhlo_instructions.cc1311 instruction_to_merge->parent()->RemoveInstruction(instruction_to_merge)); in MergeFusionInstructionIntoMultiOutput()
1318 TF_CHECK_OK(unfused_root->parent()->RemoveInstruction(unfused_root)); in MergeFusionInstructionIntoMultiOutput()
1323 TF_CHECK_OK(instruction->parent()->RemoveInstruction(instruction)); in MergeFusionInstructionIntoMultiOutput()
1497 fused_instructions_computation()->RemoveInstruction(fused_root)); in CloneAndFuseInternal()
1522 TF_CHECK_OK(parent()->RemoveInstruction(old_gte)); in CloneAndFuseInternal()
Dhlo_module.cc485 TF_CHECK_OK(computation->RemoveInstruction(*i)); in OutlineExpressionFromComputation()
Dlayout_assignment.cc2110 TF_RETURN_IF_ERROR(computation->RemoveInstruction(instruction)); in ClearPreviousPassSideEffects()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger.cc286 return computation_->RemoveInstruction(fusion); in HandleFusion()