HomeSort by relevance Sort by last modified time
    Searched refs:AddEmbeddedComputation (Results 1 - 25 of 43) sorted by null

1 2

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
while_transformer_test.cc 148 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10));
149 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1));
168 module_->AddEmbeddedComputation(BuildConditionComputation(1, 10));
169 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(1, 0, 1));
188 module_->AddEmbeddedComputation(BuildConditionComputation(0, 5));
189 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, 1));
207 module_->AddEmbeddedComputation(BuildConditionComputation(0, 10));
208 auto body = module_->AddEmbeddedComputation(BuildBodyComputation(0, 1, -1));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module_test.cc 73 module->AddEmbeddedComputation(CreateConstantComputation());
88 module->AddEmbeddedComputation(CreateConstantComputation());
90 module->AddEmbeddedComputation(CreateCallComputation({computation1}));
92 module->AddEmbeddedComputation(CreateCallComputation({computation1}));
118 fused_computation = module->AddEmbeddedComputation(b.Build());
154 module->AddEmbeddedComputation(CreateConstantComputation());
156 module->AddEmbeddedComputation(CreateCallComputation({computation1}));
158 module->AddEmbeddedComputation(CreateCallComputation({computation1}));
call_inliner_test.cc 57 module->AddEmbeddedComputation(inner.Build());
92 module->AddEmbeddedComputation(just_false.Build());
98 module->AddEmbeddedComputation(call_false_builder.Build());
131 module->AddEmbeddedComputation(just_false.Build());
154 auto outfeed_computation = module->AddEmbeddedComputation(outfeeder.Build());
hlo_subcomputation_unification_test.cc 73 module->AddEmbeddedComputation(CreateR0S32IdentityComputation());
75 module->AddEmbeddedComputation(CreateR0S32IdentityComputation());
110 module->AddEmbeddedComputation(CreateR0S32AdditionComputation());
112 module->AddEmbeddedComputation(CreateR0S32AdditionComputation());
150 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_5_));
152 module->AddEmbeddedComputation(CreateR1S32AdditionComputation(r1s32_3_));
197 module->AddEmbeddedComputation(builder.Build());
flatten_call_graph.cc 87 module->AddEmbeddedComputation(computation->Clone());
102 module->AddEmbeddedComputation(callee->Clone());
flatten_call_graph_test.cc 113 module->AddEmbeddedComputation(MakeConditionComputation());
115 module->AddEmbeddedComputation(MakeScalarComputation());
116 HloComputation* b_computation = module->AddEmbeddedComputation(
128 a_computation = module->AddEmbeddedComputation(builder.Build());
164 cond_computation = module->AddEmbeddedComputation(builder.Build());
206 module->AddEmbeddedComputation(MakeScalarComputation());
208 HloComputation* b_computation = module->AddEmbeddedComputation(
229 module->AddEmbeddedComputation(MakeScalarComputation());
hlo_scheduling_test.cc 54 module->AddEmbeddedComputation(cond_builder.Build());
61 module->AddEmbeddedComputation(body_builder.Build());
inliner_test.cc 63 hlo_module->AddEmbeddedComputation(std::move(max_f32));
97 hlo_module->AddEmbeddedComputation(std::move(const2_f32));
135 hlo_module->AddEmbeddedComputation(std::move(max_f32));
while_loop_simplifier_test.cc 63 condition = module->AddEmbeddedComputation(cond_builder.Build());
88 body = module->AddEmbeddedComputation(body_builder.Build());
104 return module->AddEmbeddedComputation(builder.Build());
196 condition = module().AddEmbeddedComputation(cond_builder.Build());
208 body = module().AddEmbeddedComputation(body_builder.Build());
244 body = module().AddEmbeddedComputation(body_builder.Build());
274 body = module().AddEmbeddedComputation(body_builder.Build());
297 body = module().AddEmbeddedComputation(body_builder.Build());
330 body = module().AddEmbeddedComputation(body_builder.Build());
366 condition = module().AddEmbeddedComputation(cond_builder.Build())
    [all...]
hlo_alias_analysis_test.cc 227 module_->AddEmbeddedComputation(subbuilder.Build());
266 module_->AddEmbeddedComputation(subbuilder.Build());
342 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build());
351 module_->AddEmbeddedComputation(cond_builder.Build());
436 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build());
444 module_->AddEmbeddedComputation(cond_builder.Build());
508 module_->AddEmbeddedComputation(build_cond_computation());
510 module_->AddEmbeddedComputation(build_cond_computation());
525 module_->AddEmbeddedComputation(inner_builder.Build());
542 module_->AddEmbeddedComputation(outer_builder.Build())
    [all...]
hlo_dce_test.cc 189 module->AddEmbeddedComputation(callee_builder.Build());
230 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build());
243 auto body_computation = module->AddEmbeddedComputation(body_builder.Build());
285 module->AddEmbeddedComputation(nested_callee_builder.Build());
296 module->AddEmbeddedComputation(callee_builder.Build());
337 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build());
378 auto reduce_subcomp = module->AddEmbeddedComputation(subcomp_builder.Build());
hlo_ordering_test.cc 109 module->AddEmbeddedComputation(builder_c.Build());
117 module->AddEmbeddedComputation(builder_b.Build());
123 module->AddEmbeddedComputation(builder_a.Build());
183 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build());
191 module->AddEmbeddedComputation(cond_builder.Build());
246 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build());
254 module->AddEmbeddedComputation(cond_builder.Build());
while_loop_invariant_code_motion_test.cc 56 return module->AddEmbeddedComputation(builder.Build());
78 return module().AddEmbeddedComputation(builder.Build());
137 return module().AddEmbeddedComputation(builder.Build());
191 return module().AddEmbeddedComputation(builder.Build());
230 return module().AddEmbeddedComputation(builder.Build());
264 return module().AddEmbeddedComputation(builder.Build());
305 return module().AddEmbeddedComputation(builder.Build());
350 return module().AddEmbeddedComputation(builder.Build());
399 while_body = module().AddEmbeddedComputation(builder.Build());
422 HloComputation* result = module().AddEmbeddedComputation(builder.Build())
    [all...]
call_graph_test.cc 119 module->AddEmbeddedComputation(MakeScalarComputation());
139 module->AddEmbeddedComputation(MakeScalarComputation());
168 module->AddEmbeddedComputation(MakeScalarComputation());
201 module->AddEmbeddedComputation(MakeScalarComputation());
244 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kCeil));
246 module->AddEmbeddedComputation(MakeScalarComputation(HloOpcode::kFloor));
303 module->AddEmbeddedComputation(MakeConditionComputation());
305 module->AddEmbeddedComputation(MakeScalarComputation());
306 HloComputation* b_computation = module->AddEmbeddedComputation(
318 a_computation = module->AddEmbeddedComputation(builder.Build())
    [all...]
copy_insertion_test.cc 749 auto condition = module_->AddEmbeddedComputation(
751 auto body = module_->AddEmbeddedComputation(
    [all...]
hlo_rematerialization_test.cc 220 module->AddEmbeddedComputation(cond_builder.Build());
222 HloComputation* body_computation = module->AddEmbeddedComputation(
259 module->AddEmbeddedComputation(cond_builder.Build());
261 HloComputation* body_computation = module->AddEmbeddedComputation(
294 module->AddEmbeddedComputation(cond_builder.Build());
296 HloComputation* inner_computation = module->AddEmbeddedComputation(
299 module->AddEmbeddedComputation(MakeRematerializableWhileComputation(
431 subcomputation = module->AddEmbeddedComputation(builder.Build());
547 subcomputation = module->AddEmbeddedComputation(builder.Build());
buffer_assignment_test.cc 583 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1"));
638 module->AddEmbeddedComputation(BuildMapComputationPlus1("f32+1"));
689 module->AddEmbeddedComputation(BuildWhileConditionComputation("if<4"));
691 module->AddEmbeddedComputation(BuildWhileBodyComputation("add-update"));
    [all...]
while_util.cc 39 return narrow_condition->parent()->AddEmbeddedComputation(builder.Build());
64 return narrow_body->parent()->AddEmbeddedComputation(builder.Build());
graphviz_example.cc 53 return module->AddEmbeddedComputation(builder.Build());
66 return module->AddEmbeddedComputation(builder.Build());
78 return module->AddEmbeddedComputation(builder.Build());
hlo_dataflow_analysis_test.cc 258 module_->AddEmbeddedComputation(subbuilder.Build());
307 module_->AddEmbeddedComputation(subbuilder.Build());
361 module_->AddEmbeddedComputation(subbuilder.Build());
414 module_->AddEmbeddedComputation(inner_builder.Build());
425 module_->AddEmbeddedComputation(outer_builder.Build());
489 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build());
498 module_->AddEmbeddedComputation(cond_builder.Build());
591 HloComputation* body = module_->AddEmbeddedComputation(body_builder.Build());
599 module_->AddEmbeddedComputation(cond_builder.Build());
658 module_->AddEmbeddedComputation(cond_builder.Build())
    [all...]
hlo_module.cc 108 HloComputation* HloModule::AddEmbeddedComputation(
458 HloComputation* nested_computation = AddEmbeddedComputation(
559 module->AddEmbeddedComputation(std::move(cloned_computation));
582 HloComputation* clone = AddEmbeddedComputation(computation->Clone("", this));
hlo_module.h 67 HloComputation* AddEmbeddedComputation(
hlo_verifier_test.cc 89 module->AddEmbeddedComputation(emb_builder.Build());
buffer_liveness_test.cc 403 module->AddEmbeddedComputation(embedded_builder.Build());
515 module->AddEmbeddedComputation(builder.Build());
577 module->AddEmbeddedComputation(builder.Build());
650 auto* computation = module->AddEmbeddedComputation(builder.Build());
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_copy_insertion_test.cc 78 HloComputation* body = module->AddEmbeddedComputation(body_builder.Build());
86 module->AddEmbeddedComputation(cond_builder.Build());
123 module->AddEmbeddedComputation(sub_builder.Build());

Completed in 626 milliseconds

1 2