HomeSort by relevance Sort by last modified time
    Searched refs:true_builder (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
conditional_test.cc 376 ComputationBuilder true_builder(client_, TestName() + ".true");
378 true_builder.Parameter(0, empty_tuple_, "tuple");
379 auto true_pred = true_builder.ConstantR0<bool>(true);
380 auto true_scalar = true_builder.ConstantR0<float>(12.2f);
381 auto true_array = true_builder.ConstantR1<float>({12.8f, 14.6f});
382 true_builder.Tuple({true_pred, true_scalar, true_array});
384 auto true_builder_result = true_builder.Build();
414 ComputationBuilder true_builder(client_, TestName() + ".true");
416 true_builder.Parameter(0, empty_tuple_, "tuple");
417 auto true_constant1 = true_builder.ConstantR0<float>(12.2f)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 677 auto true_builder = HloComputation::Builder(TestName() + "_TrueBranch"); local
679 auto param = true_builder.AddInstruction(
681 auto gte0 = true_builder.AddInstruction(
683 auto gte1 = true_builder.AddInstruction(
685 auto add = true_builder.AddInstruction(
687 true_builder.AddInstruction(HloInstruction::CreateTuple({add}));
690 module->AddEmbeddedComputation(true_builder.Build());
    [all...]
hlo_dataflow_analysis_test.cc 1560 auto true_builder = HloComputation::Builder(TestName() + "_true"); local
1633 auto true_builder = HloComputation::Builder(TestName() + "_true"); local
    [all...]

Completed in 80 milliseconds