| /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 | 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->BuildAndNoteError()
577 auto sub_builder = builder.CreateSubBuilder("power"); local
603 auto sub_builder = builder.CreateSubBuilder("power"); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| 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());
|
| buffer_assignment_test.cc | 1224 auto sub_builder = HloComputation::Builder(TestName() + "_sub"); local
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| 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;
[all...] |
