Home
last modified time | relevance | path

Searched refs:sub_builder (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmap_test.cc472 auto sub_builder = builder.CreateSubBuilder("ErrorAdd"); in TEST_F() local
473 auto x = Parameter(sub_builder.get(), 0, ShapeUtil::MakeShape(F32, {}), "x"); in TEST_F()
474 auto y = Parameter(sub_builder.get(), 1, ShapeUtil::MakeShape(U16, {}), "y"); in TEST_F()
476 auto error_add = sub_builder->BuildAndNoteError(); in TEST_F()
510 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local
511 auto x = Parameter(sub_builder.get(), 0, ShapeUtil::MakeShape(F32, {}), "x"); in TEST_F()
512 auto y = Parameter(sub_builder.get(), 1, ShapeUtil::MakeShape(F32, {}), "y"); in TEST_F()
514 auto power = sub_builder->BuildAndNoteError(); in TEST_F()
537 auto sub_builder = builder.CreateSubBuilder("power"); in TEST_F() local
538 auto x = Parameter(sub_builder.get(), 0, ShapeUtil::MakeShape(F32, {}), "x"); in TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_merger_test.cc289 HloComputation::Builder sub_builder("subcomp"); in TEST_F() local
292 auto p = sub_builder.AddInstruction( in TEST_F()
297 sum = sub_builder.AddInstruction( in TEST_F()
302 module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
Dmulti_output_fusion_test.cc736 HloComputation::Builder sub_builder("subcomp"); in TEST_F() local
737 auto* p0 = sub_builder.AddInstruction( in TEST_F()
739 auto* p1 = sub_builder.AddInstruction( in TEST_F()
741 sub_builder.AddInstruction( in TEST_F()
744 module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Darithmetic.cc118 auto sub_builder = outer_builder->CreateSubBuilder("minmax_func"); in CreateMinMaxComputation() local
119 XlaBuilder* b = sub_builder.get(); in CreateMinMaxComputation()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbfloat16_propagation_test.cc142 auto sub_builder = HloComputation::Builder("max"); in TEST_F() local
143 HloInstruction* p0 = sub_builder.AddInstruction( in TEST_F()
145 HloInstruction* p1 = sub_builder.AddInstruction( in TEST_F()
147 sub_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
149 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
Dhlo_dataflow_analysis_test.cc2743 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
2744 auto sub_param = sub_builder.AddInstruction( in TEST_F()
2746 auto one = sub_builder.AddInstruction( in TEST_F()
2748 auto ones = sub_builder.AddInstruction( in TEST_F()
2750 auto add = sub_builder.AddInstruction( in TEST_F()
2754 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
Dbuffer_assignment_test.cc1568 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
1569 auto sub_param = sub_builder.AddInstruction( in TEST_F()
1572 sub_builder.AddInstruction(HloInstruction::CreateTuple({sub_param})); in TEST_F()
1573 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2685 auto sub_builder = absl::make_unique<XlaBuilder>(computation_name); in CreateSubBuilder() local
2686 sub_builder->parent_builder_ = this; in CreateSubBuilder()
2687 sub_builder->die_immediately_on_error_ = this->die_immediately_on_error_; in CreateSubBuilder()
2688 return sub_builder; in CreateSubBuilder()