Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_copy_insertion_test.cc113 auto sub_builder = HloComputation::Builder("subcomputation"); in TEST_F() local
114 auto sub_param = sub_builder.AddInstruction( in TEST_F()
116 auto constant = sub_builder.AddInstruction( in TEST_F()
118 auto add = sub_builder.AddInstruction(HloInstruction::CreateBinary( in TEST_F()
120 sub_builder.AddInstruction( in TEST_F()
123 module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
/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/
Dmulti_output_fusion_test.cc595 HloComputation::Builder sub_builder("subcomp"); in TEST_F() local
596 auto* p0 = sub_builder.AddInstruction( in TEST_F()
598 auto* p1 = sub_builder.AddInstruction( in TEST_F()
600 sub_builder.AddInstruction( in TEST_F()
603 module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dtuple_points_to_analysis_test.cc1227 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
1228 auto sub_param = sub_builder.AddInstruction( in TEST_F()
1230 auto one = sub_builder.AddInstruction( in TEST_F()
1232 auto ones = sub_builder.AddInstruction( in TEST_F()
1234 auto add = sub_builder.AddInstruction( in TEST_F()
1238 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
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.cc2633 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
2634 auto sub_param = sub_builder.AddInstruction( in TEST_F()
2636 auto one = sub_builder.AddInstruction( in TEST_F()
2638 auto ones = sub_builder.AddInstruction( in TEST_F()
2640 auto add = sub_builder.AddInstruction( in TEST_F()
2644 auto sub_computation = module_->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
Dbuffer_assignment_test.cc1389 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); in TEST_F() local
1390 auto sub_param = sub_builder.AddInstruction( in TEST_F()
1393 sub_builder.AddInstruction(HloInstruction::CreateTuple({sub_param})); in TEST_F()
1394 auto sub_computation = module->AddEmbeddedComputation(sub_builder.Build()); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2639 auto sub_builder = absl::make_unique<XlaBuilder>(computation_name); in CreateSubBuilder() local
2640 sub_builder->parent_builder_ = this; in CreateSubBuilder()
2641 sub_builder->die_immediately_on_error_ = this->die_immediately_on_error_; in CreateSubBuilder()
2642 return sub_builder; in CreateSubBuilder()