| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_reassociate.cc | 95 TF_RETURN_IF_ERROR(computation->RemoveInstruction(inst)); in Run()
96 TF_RETURN_IF_ERROR(computation->RemoveInstruction(ar0)); in Run()
98 TF_RETURN_IF_ERROR(computation->RemoveInstruction(ar1)); in Run()
|
| D | hlo_domain_remover.cc | 85 TF_RETURN_IF_ERROR(computation->RemoveInstruction(operand)); in Run()
94 TF_RETURN_IF_ERROR(computation->RemoveInstruction(root)); in Run()
116 TF_RETURN_IF_ERROR(computation->RemoveInstruction(user)); in RemoveExitDomains()
|
| D | all_reduce_folder.cc | 206 TF_RETURN_IF_ERROR(computation->RemoveInstruction(ar1)); in Run()
207 TF_RETURN_IF_ERROR(computation->RemoveInstruction(ar0)); in Run()
|
| D | fusion_queue.h | 55 virtual void RemoveInstruction(HloInstruction* instruction) = 0;
|
| D | rng_expander.cc | 69 TF_RETURN_IF_ERROR(computation->RemoveInstruction(rng)); in ConvertSmallFpRngToF32Rng()
180 TF_RETURN_IF_ERROR(computation->RemoveInstruction(rng)); in ExpandInstruction()
|
| D | optimization_barrier_expander.cc | 57 TF_RETURN_IF_ERROR(inst->parent()->RemoveInstruction(inst)); in Run()
|
| D | reduce_scatter_decomposer.cc | 80 TF_RETURN_IF_ERROR(computation->RemoveInstruction(rs)); in Run()
|
| D | defuser.cc | 85 fusion_instruction->parent()->RemoveInstruction(fusion_instruction)); in Defuse()
|
| D | hlo_cse.cc | 97 TF_CHECK_OK(computation->RemoveInstruction(instruction)); in CombineConstants()
290 TF_RETURN_IF_ERROR(computation->RemoveInstruction(b)); in Run()
|
| D | map_inliner.cc | 83 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(map)); in HandleMap()
|
| D | conditional_code_motion.cc | 567 TF_CHECK_OK(cur_branch->RemoveInstruction(old_root)); in ConvertSpecialMove()
583 TF_CHECK_OK(conditional_parent->RemoveInstruction(conditional)); in ConvertSpecialMove()
603 TF_CHECK_OK(conditional_parent->RemoveInstruction(gte_hoist)); in ConvertSpecialMove()
676 TF_RETURN_IF_ERROR(conditional_parent->RemoveInstruction(user_instr)); in MoveInstructionOut()
705 TF_RETURN_IF_ERROR(computation->RemoveInstruction(instr_to_remove)); in MoveInstructionOut()
802 TF_RETURN_IF_ERROR(computation->RemoveInstruction(old_root)); in MoveInstructionIn()
877 TF_RETURN_IF_ERROR(conditional->parent()->RemoveInstruction(op)); in MoveInstructionIn()
|
| D | hlo_computation.cc | 240 CHECK(RemoveInstruction(old_instruction).ok()); in ReplaceParameter()
334 TF_RETURN_IF_ERROR(RemoveInstruction(item)); in RemoveInstructionAndUnusedOperands()
340 Status HloComputation::RemoveInstruction(HloInstruction* instruction) { in RemoveInstruction() function in xla::HloComputation
728 TF_CHECK_OK(RemoveInstruction(root)); in AppendInstructionsIntoCalledComputation()
733 TF_CHECK_OK(RemoveInstruction(instruction)); in AppendInstructionsIntoCalledComputation()
|
| D | instruction_fusion.cc | 467 void RemoveInstruction(HloInstruction* instruction) override { in RemoveInstruction() function in xla::__anonacdf9d610711::ReversePostOrderFusionQueue
655 fusion_queue->RemoveInstruction(operand); in Run()
657 TF_RETURN_IF_ERROR(computation->RemoveInstruction(operand)); in Run()
|
| D | hlo_element_type_converter.cc | 227 TF_RETURN_IF_ERROR(computation->RemoveInstruction(hlo)); in Run()
|
| D | hlo_domain_test.cc | 555 TF_EXPECT_OK(module->entry_computation()->RemoveInstruction(gte)); in TEST_F()
556 TF_EXPECT_OK(module->entry_computation()->RemoveInstruction(tuple)); in TEST_F()
|
| D | dot_merger.cc | 354 TF_RETURN_IF_ERROR(comp->RemoveInstruction(instr)); in MergeDots()
|
| D | while_util.cc | 135 TF_RETURN_IF_ERROR(containing_computation->RemoveInstruction(while_instr)); in MakeInstructionsLiveIn()
|
| D | conditional_simplifier_test.cc | 276 TF_ASSERT_OK(conditional_1->parent()->RemoveInstruction(conditional_1)); in TEST_F()
|
| D | multi_output_fusion.cc | 158 TF_CHECK_OK(computation()->RemoveInstruction(fused)); in Fuse()
|
| D | while_loop_concat_code_motion.cc | 944 TF_RETURN_IF_ERROR(body->RemoveInstruction(slice)); in RewriteLoopWithConcatGroups()
1008 TF_RETURN_IF_ERROR(body->RemoveInstruction(concat)); in RunOnLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_reduce_scatter_creator.cc | 105 TF_RETURN_IF_ERROR(computation->RemoveInstruction(ds)); in Run()
107 TF_RETURN_IF_ERROR(computation->RemoveInstruction(reshape)); in Run()
|
| D | multi_output_fusion.cc | 253 TF_CHECK_OK(computation_->RemoveInstruction(fused)); in FuseSiblings()
325 TF_CHECK_OK(computation_->RemoveInstruction(producer)); in DoMultiOutputFusion()
355 TF_CHECK_OK(computation_->RemoveInstruction(producer)); in DoMultiOutputFusion()
|
| D | fusion_merger.cc | 200 TF_RETURN_IF_ERROR(computation_->RemoveInstruction(instruction)); in FuseIntoAllUsers()
|
| D | horizontal_loop_fusion.cc | 440 TF_RETURN_IF_ERROR(comp->RemoveInstruction(dummy_root)); in CreateFusedComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | hlo_control_flow_flattening.cc | 293 TF_RETURN_IF_ERROR(computation->RemoveInstruction(recv)); in RemoveRecvDone()
332 TF_RETURN_IF_ERROR(computation->RemoveInstruction(send)); in RemoveSendDone()
|