Searched refs:sub_builder (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_copy_insertion_test.cc113 auto sub_builder = HloComputation::Builder("subcomputation"); local
114 auto sub_param = sub_builder.AddInstruction(
116 auto constant = sub_builder.AddInstruction(
118 auto add = sub_builder.AddInstruction(HloInstruction::CreateBinary(
120 sub_builder.AddInstruction(
123 module->AddEmbeddedComputation(sub_builder.Build());
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dmap_test.cc511 auto sub_builder = builder.CreateSubBuilder("ErrorAdd"); local
512 auto x = sub_builder->Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
513 auto y = sub_builder->Parameter(1, ShapeUtil::MakeShape(U16, {}), "y");
514 auto adder = sub_builder->Add(x, y);
515 auto error_add = sub_builder->BuildAndNoteError();
550 auto sub_builder = builder.CreateSubBuilder("power"); local
551 auto x = sub_builder->Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
552 auto y = sub_builder->Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
553 sub_builder->Pow(x, y);
554 auto power = sub_builder
577 auto sub_builder = builder.CreateSubBuilder("power"); local
603 auto sub_builder = builder.CreateSubBuilder("power"); local
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dliveness_util_test.cc428 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); local
429 auto sub_param = sub_builder.AddInstruction(
431 auto one = sub_builder.AddInstruction(
433 auto ones = sub_builder.AddInstruction(
435 auto add = sub_builder.AddInstruction(
439 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build());
H A Dbuffer_assignment_test.cc1224 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); local
1225 auto sub_param = sub_builder.AddInstruction(
1228 sub_builder.AddInstruction(HloInstruction::CreateTuple({sub_param}));
1229 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build());
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dcomputation_builder.cc56 auto sub_builder = MakeUnique<ComputationBuilder>(client_, computation_name); local
57 sub_builder->parent_builder_ = this;
58 sub_builder->die_immediately_on_error_ = die_immediately_on_error_;
59 return sub_builder;

Completed in 163 milliseconds