| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| 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/ |
| map_test.cc | 472 auto sub_builder = builder.CreateSubBuilder("ErrorAdd"); local
473 auto x = Parameter(sub_builder.get(), 0, ShapeUtil::MakeShape(F32, {}), "x");
474 auto y = Parameter(sub_builder.get(), 1, ShapeUtil::MakeShape(U16, {}), "y");
476 auto error_add = sub_builder->BuildAndNoteError();
510 auto sub_builder = builder.CreateSubBuilder("power"); local
511 auto x = Parameter(sub_builder.get(), 0, ShapeUtil::MakeShape(F32, {}), "x");
512 auto y = Parameter(sub_builder.get(), 1, ShapeUtil::MakeShape(F32, {}), "y");
514 auto power = sub_builder->BuildAndNoteError();
537 auto sub_builder = builder.CreateSubBuilder("power"); local
538 auto x = Parameter(sub_builder.get(), 0, ShapeUtil::MakeShape(F32, {}), "x")
563 auto sub_builder = builder.CreateSubBuilder("power"); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| bfloat16_propagation_test.cc | 142 auto sub_builder = HloComputation::Builder("max"); local
143 HloInstruction* p0 = sub_builder.AddInstruction(
145 HloInstruction* p1 = sub_builder.AddInstruction(
147 sub_builder.AddInstruction(HloInstruction::CreateBinary(
149 auto max_computation = module->AddEmbeddedComputation(sub_builder.Build());
[all...] |
| tuple_points_to_analysis_test.cc | 1227 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); local
[all...] |
| hlo_dataflow_analysis_test.cc | 2633 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); local
[all...] |
