Home
last modified time | relevance | path

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

123

/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/
Dhlo_module_test.cc78 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
93 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
95 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
97 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
123 fused_computation = module->AddEmbeddedComputation(b.Build()); in TEST_F()
159 module->AddEmbeddedComputation(CreateConstantComputation()); in TEST_F()
161 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
163 module->AddEmbeddedComputation(CreateCallComputation({computation1})); in TEST_F()
289 module->AddEmbeddedComputation(reduction->Clone()); in TEST_F()
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_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.cc760 auto condition = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
762 auto body = module_->AddEmbeddedComputation( in BuildWhileInstructionWithCustomInit()
794 auto condition = module_->AddEmbeddedComputation( in TEST_F()
797 module_->AddEmbeddedComputation(BuildIndependentBodyComputation()); in TEST_F()
823 auto condition = module_->AddEmbeddedComputation( in TEST_F()
825 auto body = module_->AddEmbeddedComputation(BuildDependentBodyComputation()); in TEST_F()
867 auto condition = module_->AddEmbeddedComputation( in TEST_F()
869 auto body = module_->AddEmbeddedComputation( in TEST_F()
883 auto condition1 = module_->AddEmbeddedComputation( in TEST_F()
885 auto condition2 = module_->AddEmbeddedComputation( in TEST_F()
[all …]
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()
156 auto outfeed_computation = module->AddEmbeddedComputation(outfeeder.Build()); in TEST_F()
Dhlo_alias_analysis_test.cc350 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
359 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
421 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
461 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_F()
538 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
547 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
633 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
641 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
706 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F()
708 module_->AddEmbeddedComputation(build_cond_computation()); in TEST_F()
[all …]
Dhlo_rematerialization_test.cc236 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
238 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F()
272 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
274 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F()
304 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
306 HloComputation* inner_computation = module->AddEmbeddedComputation( in TEST_F()
309 module->AddEmbeddedComputation(MakeRematerializableWhileComputation( in TEST_F()
436 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
595 subcomputation = module->AddEmbeddedComputation(builder.Build()); in TEST_P()
Dhlo_ordering_test.cc63 module->AddEmbeddedComputation(builder_c.Build()); in TEST_F()
71 module->AddEmbeddedComputation(builder_b.Build()); in TEST_F()
77 module->AddEmbeddedComputation(builder_a.Build()); in TEST_F()
137 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
145 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
220 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
228 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
468 HloComputation* subcomputation = module->AddEmbeddedComputation( in TEST_F()
Dbuffer_assignment_test.cc746 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1")); in TEST_F()
801 module->AddEmbeddedComputation(BuildReduceComputation("f32+f32")); in TEST_F()
852 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4")); in TEST_F()
854 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update")); in TEST_F()
917 auto true_computation = module->AddEmbeddedComputation( in TEST_F()
919 auto false_computation = module->AddEmbeddedComputation( in TEST_F()
1208 auto map_computation = module->AddEmbeddedComputation(map_builder.Build()); in TEST_F()
1216 auto call_computation = module->AddEmbeddedComputation(call_builder.Build()); in TEST_F()
1394 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
1460 module->AddEmbeddedComputation(std::move(d_computation)); in TEST_F()
[all …]
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_dce_test.cc188 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F()
230 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
243 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
287 module->AddEmbeddedComputation(nested_callee_builder.Build()); in TEST_F()
298 module->AddEmbeddedComputation(callee_builder.Build()); in TEST_F()
339 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
380 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build()); in TEST_F()
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()
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()
396 auto comp_f0 = module->AddEmbeddedComputation(builder_f0.Build()); in TEST_F()
409 auto comp_f1 = module->AddEmbeddedComputation(builder_f1.Build()); in TEST_F()
446 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
485 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
590 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()
597 auto body = module->AddEmbeddedComputation(builder_body.Build()); in TEST_F()
648 auto cond = module->AddEmbeddedComputation(builder_cond.Build()); in TEST_F()
660 auto comp_f = module->AddEmbeddedComputation(builder_f.Build()); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc262 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
312 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
367 module_->AddEmbeddedComputation(subbuilder.Build()); in TEST_P()
421 module_->AddEmbeddedComputation(inner_builder.Build()); in TEST_P()
432 module_->AddEmbeddedComputation(outer_builder.Build()); in TEST_P()
497 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P()
506 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
600 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_P()
608 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
668 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_P()
[all …]
Dflatten_call_graph.cc90 module->AddEmbeddedComputation(computation->Clone()); in FlattenNode()
105 module->AddEmbeddedComputation(callee->Clone()); in FlattenNode()
Dwhile_loop_simplifier.cc265 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryRemoveDeadWhileParams()
266 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveDeadWhileParams()
440 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryRemoveConstantParams()
441 module->AddEmbeddedComputation(std::move(new_while_body)), in TryRemoveConstantParams()
732 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryFlattenNestedTuples()
733 module->AddEmbeddedComputation(std::move(new_while_body)), in TryFlattenNestedTuples()
946 module->AddEmbeddedComputation(while_body->CloneWithReplacementPairs({ in TryMergeInductionVariables()
967 module->AddEmbeddedComputation(std::move(new_while_cond)), in TryMergeInductionVariables()
968 module->AddEmbeddedComputation(std::move(new_while_body)), in TryMergeInductionVariables()
Dwhile_util.cc46 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileCondition()
70 return narrow_body->parent()->AddEmbeddedComputation(builder.Build()); in WidenWhileBody()
246 loop_state_shape, module->AddEmbeddedComputation(std::move(cond)), in MakeCountedLoop()
247 module->AddEmbeddedComputation(std::move(body)), init_tuple)); in MakeCountedLoop()
Dconditional_simplifier_test.cc64 module->AddEmbeddedComputation(true_computation_builder.Build()); in MakeConditional()
81 module->AddEmbeddedComputation(false_computation_builder.Build()); in MakeConditional()
Dhlo_module.cc119 HloComputation* HloModule::AddEmbeddedComputation( in AddEmbeddedComputation() function in xla::HloModule
470 HloComputation* nested_computation = AddEmbeddedComputation( in OutlineExpressionFromComputation()
599 AddEmbeddedComputation(computation->Clone(context->suffix(), context)); in DeepCloneComputation()
601 new_computation = AddEmbeddedComputation(computation->Clone("")); in DeepCloneComputation()
Ddynamic_dimension_inference_test.cc61 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetAdd()
72 return module_->AddEmbeddedComputation(embedded_builder.Build()); in GetGe()
485 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
493 module_->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
Dtuple_simplifier_test.cc243 c0 = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
258 c1 = module->AddEmbeddedComputation(builder.Build()); in TEST_F()
Dbuffer_liveness_test.cc405 module->AddEmbeddedComputation(embedded_builder.Build()); in TEST_F()
519 module->AddEmbeddedComputation(builder.Build()); in TEST_F()
581 module->AddEmbeddedComputation(builder.Build()); in TEST_F()
807 module->AddEmbeddedComputation(builder.Build()); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc78 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F()
86 module->AddEmbeddedComputation(cond_builder.Build()); in TEST_F()
123 module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()

123