HomeSort by relevance Sort by last modified time
    Searched defs:cond_builder (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/lib/
while_loop.cc 48 std::unique_ptr<xla::ComputationBuilder> cond_builder = local
51 auto parameter = cond_builder->Parameter(0, tuple_shape, "parameter");
55 condition_function(unpack_tuple(parameter, arity, cond_builder.get()),
56 cond_builder.get()));
57 TF_RETURN_IF_ERROR(cond_builder->SetReturnValue(result));
59 TF_ASSIGN_OR_RETURN(auto cond, cond_builder->Build());
88 xla::ComputationBuilder* cond_builder)
90 return cond_builder->Lt(
92 IntegerLiteral(cond_builder, num_iterations_type, num_iterations));
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_copy_insertion_test.cc 80 auto cond_builder = HloComputation::Builder("condition"); local
81 cond_builder.AddInstruction(
83 cond_builder.AddInstruction(
86 module->AddEmbeddedComputation(cond_builder.Build());
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_scheduling_test.cc 41 auto cond_builder = HloComputation::Builder("WhileCond"); local
43 HloInstruction* cond_param = cond_builder.AddInstruction(
45 HloInstruction* cond_iter = cond_builder.AddInstruction(
47 HloInstruction* cond_data = cond_builder.AddInstruction(
50 HloInstruction* cond_lt = cond_builder.AddInstruction(
54 module->AddEmbeddedComputation(cond_builder.Build());
hlo_dce_test.cc 221 auto cond_builder = HloComputation::Builder(TestName() + "-cond"); local
223 auto param = cond_builder.AddInstruction(
225 auto constant = cond_builder.AddInstruction(
227 cond_builder.AddInstruction(HloInstruction::CreateBinary(
230 auto cond_computation = module->AddEmbeddedComputation(cond_builder.Build());
hlo_ordering_test.cc 185 auto cond_builder = HloComputation::Builder("condition"); local
186 auto cond_param = cond_builder.AddInstruction(
188 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert(
191 module->AddEmbeddedComputation(cond_builder.Build());
248 auto cond_builder = HloComputation::Builder("condition"); local
249 auto cond_param = cond_builder.AddInstruction(
251 auto convert = cond_builder.AddInstruction(HloInstruction::CreateConvert(
254 module->AddEmbeddedComputation(cond_builder.Build());
hlo_rematerialization_test.cc 214 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); local
215 cond_builder.AddInstruction(
217 cond_builder.AddInstruction(
220 module->AddEmbeddedComputation(cond_builder.Build());
253 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); local
254 cond_builder.AddInstruction(
256 cond_builder.AddInstruction(
259 module->AddEmbeddedComputation(cond_builder.Build());
288 auto cond_builder = HloComputation::Builder(TestName() + ".cond"); local
289 cond_builder.AddInstruction
    [all...]
heap_simulator_test.cc 420 auto cond_builder = HloComputation::Builder("WhileCond"); local
421 HloInstruction* cond_param = cond_builder.AddInstruction(
423 HloInstruction* cond_iter = cond_builder.AddInstruction(
425 HloInstruction* cond_data = cond_builder.AddInstruction(
427 HloInstruction* cond_lt = cond_builder.AddInstruction(
431 tracker.module()->AddEmbeddedComputation(cond_builder.Build());
    [all...]
hlo_alias_analysis_test.cc 345 auto cond_builder = HloComputation::Builder("condition"); local
346 auto cond_param = cond_builder.AddInstruction(
348 cond_builder.AddInstruction(
351 module_->AddEmbeddedComputation(cond_builder.Build());
438 auto cond_builder = HloComputation::Builder("condition"); local
439 cond_builder.AddInstruction(
441 cond_builder.AddInstruction(
444 module_->AddEmbeddedComputation(cond_builder.Build());
497 auto cond_builder = HloComputation::Builder("condition");
498 cond_builder.AddInstruction
607 auto cond_builder = HloComputation::Builder("condition"); local
753 auto cond_builder = HloComputation::Builder("condition"); local
849 auto cond_builder = HloComputation::Builder("condition"); local
    [all...]
copy_insertion_test.cc 1304 auto cond_builder = HloComputation::Builder("condition"); local
1369 auto cond_builder = HloComputation::Builder("condition"); local
1429 auto cond_builder = HloComputation::Builder("condition"); local
1514 auto cond_builder = HloComputation::Builder("condition"); local
1576 auto cond_builder = HloComputation::Builder("condition"); local
    [all...]
hlo_dataflow_analysis_test.cc 492 auto cond_builder = HloComputation::Builder("condition"); local
493 auto cond_param = cond_builder.AddInstruction(
495 auto cond_constant = cond_builder.AddInstruction(
498 module_->AddEmbeddedComputation(cond_builder.Build());
593 auto cond_builder = HloComputation::Builder("condition"); local
594 cond_builder.AddInstruction(
596 cond_builder.AddInstruction(
599 module_->AddEmbeddedComputation(cond_builder.Build());
652 auto cond_builder = HloComputation::Builder("condition"); local
653 cond_builder.AddInstruction
779 auto cond_builder = HloComputation::Builder("condition"); local
1025 auto cond_builder = HloComputation::Builder("condition"); local
1319 auto cond_builder = HloComputation::Builder("condition"); local
    [all...]

Completed in 216 milliseconds