| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| fusion_test.cc | 471 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( local
474 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, reverse1},
488 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( local
491 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, broadcast1},
505 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( local
508 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, slice1},
543 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( local
546 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, reshape1},
561 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( local
564 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, transpose1}
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_cse_test.cc | 341 // negate2 = neg(constant)
343 // add2 = add(negate2, exp2)
358 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( local
363 constant->shape(), HloOpcode::kAdd, negate2, exp2));
372 EXPECT_THAT(tuple, op::Tuple(op::Add(negate1, exp1), op::Add(negate2, exp2)));
|
| hlo_computation_test.cc | 135 auto negate2 = builder.AddInstruction( local
140 ElementsAre(constant, negate1, negate2));
152 auto negate2 = builder.AddInstruction( local
158 ElementsAre(constant, negate1, negate2, trace));
|
| layout_assignment_test.cc | 150 auto negate2 = builder.AddInstruction( local
157 {negate2, negate1, add}, HloInstruction::FusionKind::kLoop);
[all...] |
