Searched refs:sub_builder (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_copy_insertion_test.cc | 113 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 D | map_test.cc | 511 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 D | liveness_util_test.cc | 428 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 D | buffer_assignment_test.cc | 1224 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 D | computation_builder.cc | 56 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