Home
last modified time | relevance | path

Searched refs:AddEmbeddedComputation (Results 1 – 25 of 67) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_test.cc79 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
94 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
96 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
98 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
124 fused_computation = module->AddEmbeddedComputation(b.Build()); in TEST_F()
160 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
162 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
164 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
287 module->AddEmbeddedComputation(reduction->Clone()); in TEST_F()
380 new_comp = module->AddEmbeddedComputation(b.Build()); in TEST_F()
[all …]
Dhlo_subcomputation_unification_test.cc72 module->AddEmbeddedComputation(CreateR0S32IdentityComputation()); in TEST_F()
74 module->AddEmbeddedComputation(CreateR0S32IdentityComputation()); in TEST_F()
99 module->AddEmbeddedComputation(CreateR0S32AdditionComputation()); in TEST_F()
101 module->AddEmbeddedComputation(CreateR0S32AdditionComputation()); in TEST_F()
129 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_5_)); in TEST_F()
131 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_3_)); in TEST_F()
166 module->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dflatten_call_graph_test.cc114 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
116 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
117 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F()
129 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
165 cond_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
207 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
209 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F()
230 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
Dcall_inliner_test.cc56 module->AddEmbeddedComputation(inner.Build()); in TEST_F()
91 module->AddEmbeddedComputation(just_false.Build()); in TEST_F()
99 module->AddEmbeddedComputation(call_false_builder.Build()); in TEST_F()
132 module->AddEmbeddedComputation(just_false.Build()); in TEST_F()
155 module->AddEmbeddedComputation(empty.Build()); in TEST_F()
162 module->AddEmbeddedComputation(empty2.Build()); in TEST_F()
196 auto outfeed_computation = module->AddEmbeddedComputation(outfeeder.Build()); in TEST_F()
Dcall_graph_test.cc121 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
143 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
174 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
207 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
251 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kCeil)); in TEST_F()
253 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor)); in TEST_F()
313 module->AddEmbeddedComputation(MakeConditionComputation()); in TEST_F()
315 module->AddEmbeddedComputation(MakeScalarComputation()); in TEST_F()
316 HloComputation* b_computation = module->AddEmbeddedComputation( in TEST_F()
328 a_computation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
[all …]
Dcopy_insertion_test.cc766 auto condition = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
768 auto body = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
800 auto condition = module_->AddEmbeddedComputation( in TEST_F()
803 module_->AddEmbeddedComputation(BuildIndependentBodyComputation()); in TEST_F()
862 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
864 module_->AddEmbeddedComputation(while_hlo->while_body()->Clone()); in TEST_F()
930 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
932 module_->AddEmbeddedComputation(while_hlo->while_body()->Clone()); in TEST_F()
990 module_->AddEmbeddedComputation(while_hlo->while_condition()->Clone()); in TEST_F()
992 module_->AddEmbeddedComputation(while_hlo->while_body()->Clone()); in TEST_F()
[all …]
Dhlo_ordering_test.cc62 module->AddEmbeddedComputation(builder_c.Build()); in TEST_F()
70 module->AddEmbeddedComputation(builder_b.Build()); in TEST_F()
76 module->AddEmbeddedComputation(builder_a.Build()); in TEST_F()
136 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
144 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
219 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
227 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
474 HloComputation* subcomputation = module->AddEmbeddedComputation( in TEST_F()
Dwhile_loop_invariant_code_motion_test.cc57 return module->AddEmbeddedComputation(builder.Build()); in MakeAlwaysTrueComputation()
80 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
139 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
193 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
233 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
273 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
326 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
378 return m->AddEmbeddedComputation(builder.Build()); in TEST_F()
427 while_body = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
451 HloComputation* result = m->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dhlo_alias_analysis_test.cc344 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
353 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
413 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
453 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
530 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
539 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
625 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
633 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
698 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F()
700 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F()
[all …]
Dmap_inliner_test.cc63 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F()
97 hlo_module->AddEmbeddedComputation(std::move(const2_f32)); in TEST_F()
135 hlo_module->AddEmbeddedComputation(std::move(max_f32)); in TEST_F()
167 hlo_module->AddEmbeddedComputation(std::move(param_f32)); in TEST_F()
Dhlo_dce_test.cc226 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F()
268 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
283 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
327 module->AddEmbeddedComputation(nested_callee_builder.Build()); in TEST_F()
338 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F()
377 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
418 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
Dbuffer_assignment_test.cc793 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
816 module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
918 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1")); in TEST_F()
973 module->AddEmbeddedComputation(BuildReduceComputation("f32+f32")); in TEST_F()
1024 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()
1026 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()
1089 auto true_computation = module->AddEmbeddedComputation( in TEST_F()
1091 auto false_computation = module->AddEmbeddedComputation( in TEST_F()
1380 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F()
1388 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F()
[all …]
Dhlo_rematerialization_test.cc156 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
158 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F()
192 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
194 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F()
224 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
226 module->AddEmbeddedComputation(while_cond->Clone()); in TEST_F()
228 HloComputation* inner_computation = module->AddEmbeddedComputation( in TEST_F()
231 module->AddEmbeddedComputation(MakeRematerializableWhileComputation( in TEST_F()
358 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
515 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()
Dbfloat16_propagation_test.cc149 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
210 auto reduction = module->AddEmbeddedComputation(rb.Build()); in TEST_F()
397 auto comp_f0 = module->AddEmbeddedComputation(builder_f0.Build()); in TEST_F()
410 auto comp_f1 = module->AddEmbeddedComputation(builder_f1.Build()); in TEST_F()
442 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
476 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
515 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
620 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()
627 auto body = module->AddEmbeddedComputation(builder_body.Build()); in TEST_F()
678 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc265 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
328 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_P()
339 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_P()
404 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P()
413 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
507 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P()
515 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
574 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
585 module_->AddEmbeddedComputation(level0_builder.Build()); in TEST_P()
594 module_->AddEmbeddedComputation(level1_builder.Build()); in TEST_P()
[all …]
Dflatten_call_graph.cc90 module->AddEmbeddedComputation(computation->Clone()); in FlattenNode()
105 module->AddEmbeddedComputation(callee->Clone()); in FlattenNode()
Dwhile_util.cc46 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileCondition()
70 return narrow_body->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileBody()
261 loop_state_shape, module->AddEmbeddedComputation(std::move(cond)), in MakeCountedLoop()
262 module->AddEmbeddedComputation(std::move(body)), init_tuple)); in MakeCountedLoop()
Dwhile_loop_simplifier.cc162 module->AddEmbeddedComputation(std::move(new_while_cond)), in RemoveDeadTupleIndices()
163 module->AddEmbeddedComputation(std::move(new_while_body)), in RemoveDeadTupleIndices()
621 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryRemoveConstantParams()
622 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveConstantParams()
944 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryFlattenNestedTuples()
945 module->AddEmbeddedComputation(std::move(new_while_body)), in TryFlattenNestedTuples()
1158 module->AddEmbeddedComputation(while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()
1179 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryMergeInductionVariables()
1180 module->AddEmbeddedComputation(std::move(new_while_body)), in TryMergeInductionVariables()
Ddynamic_dimension_inference_test.cc64 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAdd()
84 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAddTuple()
95 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetGe()
753 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
761 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
844 module_->AddEmbeddedComputation(true_builder.Build()); in TEST_F()
865 module_->AddEmbeddedComputation(false_builder.Build()); in TEST_F()
1134 module_->AddEmbeddedComputation(compare_builder.Build()); in TEST_F()
1172 module_->AddEmbeddedComputation(compare_builder.Build()); in TEST_F()
Dconditional_simplifier.cc242 computation->parent()->AddEmbeddedComputation(computation->Clone()); in TryRemoveUnusedConditionalOperands()
302 conditional_op->GetModule()->AddEmbeddedComputation( in ReplaceRootWithEmptyTupleIfNoUsers()
417 conditional_op->GetModule()->AddEmbeddedComputation( in RemoveUnusedTupleElements()
Dtuple_simplifier_test.cc243 c0 = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
258 c1 = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dhlo_module.cc151 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation() function in xla::HloModule
571 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation()
766 AddEmbeddedComputation(computation->Clone(context->suffix(), context)); in DeepCloneComputation()
768 new_computation = AddEmbeddedComputation(computation->Clone("")); in DeepCloneComputation()
Dheap_simulator_test.cc61 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
68 module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
151 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
164 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
527 module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
601 module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
651 tracker.module()->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()
903 tracker.module()->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
909 tracker.module()->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc128 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
129 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
139 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10)); in TEST_F()
140 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(1, 0, 1)); in TEST_F()
150 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5)); in TEST_F()
151 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1)); in TEST_F()
161 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10)); in TEST_F()
162 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, -1)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dparallel_reduction_test.cc92 module->AddEmbeddedComputation(embedded_builder.Build()); in TEST_F()
112 fusion_computation = module->AddEmbeddedComputation(fusion_builder.Build()); in TEST_F()

123