Searched refs:set_root_instruction (Results 1 – 17 of 17) sorted by relevance
56 entry_computation->set_root_instruction(first_1_dims_collapsed); in TEST_F()75 entry_computation->set_root_instruction(first_2_dims_collapsed); in TEST_F()98 entry_computation->set_root_instruction(with_1_degenerate_dim_prepended); in TEST_F()117 entry_computation->set_root_instruction(with_2_degenerate_dims_prepended); in TEST_F()136 entry_computation->set_root_instruction(with_2_degenerate_dims_prepended); in TEST_F()155 entry_computation->set_root_instruction(first_dim_expanded); in TEST_F()177 entry_computation->set_root_instruction(zero_padded_param); in TEST_F()196 entry_computation->set_root_instruction(zeros); in TEST_F()215 entry_computation->set_root_instruction(zeros); in TEST_F()
103 entry->set_root_instruction(sub); in TEST_F()154 entry->set_root_instruction(new_root); in TEST_F()196 entry->set_root_instruction(new_root); in TEST_F()258 cond->set_root_instruction(cond->AddInstruction( in TEST_F()264 body->set_root_instruction(body->parameter_instruction(0)); in TEST_F()
56 wide_while_cond->set_root_instruction(call_narrow_cond); in WidenWhileCondition()88 wide_while_body->set_root_instruction( in WidenWhileBody()170 cond_computation->set_root_instruction(compare); in MakeCountedLoopConditionComputation()200 body_computation->set_root_instruction(next_state_tuple); in MakeCountedLoopBodyComputation()
91 computation->set_root_instruction(root->mutable_operand(0)); in Run()
309 body->set_root_instruction(root_copy); in AddCopiesForWhile()331 computation->set_root_instruction(deep_copy); in AddCopiesForConditional()416 entry->set_root_instruction(root_copied); in AddCopiesForAliasedInputOutputs()1085 instruction->parent()->set_root_instruction(deep_copy); in AddSpecialCaseCopies()1276 computation->set_root_instruction(root_copy); in AddCopiesForBufferAssignment()
95 computation->set_root_instruction(convert); in InsertConvertAfterOutput()263 computation_->set_root_instruction(tuple); in HandleMultipleOutputs()
194 comparator->set_root_instruction(new_root); in ExpandInstruction()
655 fusion_computation->set_root_instruction(copy); in ResolveInconsistentFusions()702 computation->set_root_instruction(source); in SkipNoopConversions()
148 void set_root_instruction(HloInstruction* new_root_instruction,
294 entry->set_root_instruction(negate); in TEST_F()
132 pass_through.operand->parent()->set_root_instruction(gte); in FixupPassThroughDomainLinks()
275 void HloComputation::set_root_instruction(HloInstruction* new_root_instruction, in set_root_instruction() function in xla::HloComputation602 set_root_instruction(fusion_instruction); in FuseInstructionsInto()
461 module->entry_computation()->set_root_instruction(tuple->mutable_operand(0)); in TEST_F()
1803 computation->set_root_instruction(new_root); in RunOnComputation()
748 xla_while->parent()->set_root_instruction(tuple); in BuildWhileInstruction_InitPointsToInterfering()
1493 fused_instructions_computation()->set_root_instruction(new_root); in CloneAndFuseInternal()
1981 parent_->set_root_instruction(new_producer, in ReplaceAllUsesWithDifferentShape()