Home
last modified time | relevance | path

Searched refs:set_root_instruction (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils_test.cc71 entry_computation->set_root_instruction(first_1_dims_collapsed); in TEST_F()
90 entry_computation->set_root_instruction(first_2_dims_collapsed); in TEST_F()
113 entry_computation->set_root_instruction(with_1_degenerate_dim_prepended); in TEST_F()
132 entry_computation->set_root_instruction(with_2_degenerate_dims_prepended); in TEST_F()
151 entry_computation->set_root_instruction(with_2_degenerate_dims_prepended); in TEST_F()
170 entry_computation->set_root_instruction(first_dim_expanded); in TEST_F()
192 entry_computation->set_root_instruction(zero_padded_param); in TEST_F()
211 entry_computation->set_root_instruction(zeros); in TEST_F()
230 entry_computation->set_root_instruction(zeros); in TEST_F()
252 entry_computation->set_root_instruction(output); in TEST_F()
[all …]
Dhlo_schedule_test.cc102 entry->set_root_instruction(sub); in TEST_F()
153 entry->set_root_instruction(new_root); in TEST_F()
195 entry->set_root_instruction(new_root); in TEST_F()
257 cond->set_root_instruction(cond->AddInstruction( in TEST_F()
263 body->set_root_instruction(body->parameter_instruction(0)); in TEST_F()
Droot_instruction_sinker.cc35 computation->set_root_instruction(new_root); in SinkTupleRoot()
47 computation->set_root_instruction(new_root); in SinkNontupleRoot()
Dwhile_util.cc56 wide_while_cond->set_root_instruction(call_narrow_cond); in WidenWhileCondition()
88 wide_while_body->set_root_instruction( in WidenWhileBody()
172 cond_computation->set_root_instruction(compare); in MakeCountedLoopConditionComputation()
202 body_computation->set_root_instruction(next_state_tuple); in MakeCountedLoopBodyComputation()
Dconditional_canonicalizer.cc32 branch->set_root_instruction(tuple, /*accept_different_shape=*/true); in CanonicalizeNonTupleConditional()
Dhlo_domain_remover.cc91 computation->set_root_instruction(root->mutable_operand(0)); in Run()
Dbfloat16_normalization.cc164 computation->set_root_instruction(new_hlo, /*accept_different_shape=*/true); in InsertConvertAfterOutput()
212 computation->set_root_instruction(new_hlo, /*accept_different_shape=*/true); in ChangeOutputTypeThenInsertConvertBack()
377 computation_->set_root_instruction(tuple); in HandleMultipleOutputs()
Dcopy_insertion.cc331 body->set_root_instruction(root_copy); in AddCopiesForWhile()
428 entry->set_root_instruction(root_copied); in AddCopiesForAliasedInputOutputs()
1005 computation->set_root_instruction(deep_copy); in AddCopiesForConditional()
1159 instruction->parent()->set_root_instruction(deep_copy); in AddSpecialCaseCopies()
Dconditional_code_motion.cc370 computation->set_root_instruction(new_tuple); in RestructureConditionalInstruction()
480 cur_branch->set_root_instruction(new_branch_root, true /*new shape*/); in ConvertSpecialMove()
605 computation->set_root_instruction(tuple, true); in MoveInstructionOut()
708 computation->set_root_instruction(new_root, in MoveInstructionIn()
768 conditional->parent()->set_root_instruction(gtr); in MoveInstructionIn()
Dconditional_simplifier.cc307 branch_computation->set_root_instruction(new_empty_root, in ReplaceRootWithEmptyTupleIfNoUsers()
432 cloned_branch->set_root_instruction(new_tuple_root, in RemoveUnusedTupleElements()
Dbfloat16_propagation.cc773 fusion_computation->set_root_instruction(copy); in ResolveInconsistentFusions()
822 computation->set_root_instruction(source); in SkipNoopConversions()
944 inst->parent()->set_root_instruction(copy, in Run()
Dstable_sort_expander.cc194 comparator->set_root_instruction(new_root); in ExpandInstruction()
Ddynamic_padder.cc695 comp->set_root_instruction(gather); in RewriteDynamicReshapeCombineInput()
1258 sort_comp->set_root_instruction(new_root); in RewriteDynamicSort()
1278 comp->set_root_instruction(rewritten_sort); in RewriteDynamicSort()
1537 module->entry_computation()->set_root_instruction(static_param); in InsertPadToStaticAfterModuleInputs()
1589 computation->set_root_instruction(new_root); in Run()
Dhlo_sharding_metadata.cc132 pass_through.operand->parent()->set_root_instruction(gte); in FixupPassThroughDomainLinks()
Dhlo_computation.h185 void set_root_instruction(HloInstruction* new_root_instruction,
Dhlo_module_test.cc292 entry->set_root_instruction(negate); in TEST_F()
Ddynamic_dimension_inference.cc62 wide_comp->set_root_instruction(call_narrow_comp, in WidenComputation()
1361 new_computation->set_root_instruction(new_branch_root, in HandleConditional()
1533 hlo->while_body()->set_root_instruction(new_body_root); in HandleWhile()
Dhlo_computation.cc330 void HloComputation::set_root_instruction(HloInstruction* new_root_instruction, in set_root_instruction() function in xla::HloComputation
730 set_root_instruction(fusion_instruction); in FuseInstructionsInto()
Dhlo_domain_test.cc461 module->entry_computation()->set_root_instruction(tuple->mutable_operand(0)); in TEST_F()
Dlayout_assignment.cc892 branch_comp->set_root_instruction(param_copy, in CopyOperandIfLayoutsDiffer()
1950 computation->set_root_instruction(new_root); in RunOnComputation()
Dhlo_instructions.cc1508 computation->set_root_instruction(dummy_root, in MergeFusionInstruction()
1762 fused_instructions_computation()->set_root_instruction(new_root); in CloneAndFuseInternal()
Dhlo_instruction.cc2375 parent_->set_root_instruction(new_producer, in ReplaceAllUsesWithDifferentShape()
2414 parent_->set_root_instruction(new_producer, in ReplaceAllUsesWithDifferentShape()
Dcopy_insertion_test.cc754 xla_while->parent()->set_root_instruction(tuple); in BuildWhileInstruction_InitPointsToInterfering()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcublas_gemm_pad_for_tensor_cores.cc111 parent->set_root_instruction(slice); in PadForTensorCores()
Dhorizontal_loop_fusion.cc420 comp->set_root_instruction(tuple, /*accept_different_shape=*/true); in CreateFusedComputation()

12