| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_computation_test.cc | 140 auto negate1 = builder.AddInstruction( local
143 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1));
147 ElementsAre(constant, negate1, negate2));
155 auto negate1 = builder.AddInstruction( local
158 builder.AddInstruction(HloInstruction::CreateTrace("foobar", negate1));
160 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1));
165 ElementsAre(constant, negate1, negate2, trace));
|
| hlo_cse_test.cc | 527 // negate1 = neg(constant)
529 // add1 = add(negate1, exp1)
540 auto negate1 = builder.AddInstruction(HloInstruction::CreateUnary( local
545 constant->shape(), HloOpcode::kAdd, negate1, exp1));
561 EXPECT_THAT(tuple, op::Tuple(op::Add(negate1, exp1), op::Add(negate2, exp2)));
|
| copy_insertion_test.cc | 1519 auto negate1 = builder.AddInstruction( local
1558 auto negate1 = builder.AddInstruction( local
[all...] |
| layout_assignment_test.cc | 177 auto negate1 = builder.AddInstruction( local
180 HloInstruction::CreateUnary(ashape, HloOpcode::kNegate, negate1));
186 {negate2, negate1, add}, HloInstruction::FusionKind::kLoop);
[all...] |
| hlo_alias_analysis_test.cc | 235 auto negate1 = builder.AddInstruction( local
239 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1}));
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| cpu_fusion_test.cc | 288 // negate1 = negate(exp1)
291 // tuple = tuple(negate1, negate2, exp2)
293 // exp1 should be fused down into negate1, but exp2 will not be fused into
306 auto negate1 = builder.AddInstruction( local
315 HloInstruction::CreateTuple({negate1, negate2, exp2}));
324 // negate1).
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| instruction_fusion_test.cc | 58 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( local
62 ShapeUtil::MakeShape(S32, {1}), negate1, {0}));
|
