Searched defs:negate1 (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | instruction_fusion_test.cc | 53 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( local 57 ShapeUtil::MakeShape(S32, {1}), negate1, {0}));
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_cse_test.cc | 338 // negate1 = neg(constant) 340 // add1 = add(negate1, exp1) 351 auto negate1 = builder.AddInstruction(HloInstruction::CreateUnary( local 356 constant->shape(), HloOpcode::kAdd, negate1, exp1)); 372 EXPECT_THAT(tuple, op::Tuple(op::Add(negate1, exp1), op::Add(negate2, exp2)));
|
H A D | hlo_computation_test.cc | 133 auto negate1 = builder.AddInstruction( local 136 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); 140 ElementsAre(constant, negate1, negate2)); 148 auto negate1 = builder.AddInstruction( local 151 builder.AddInstruction(HloInstruction::CreateTrace("foobar", negate1)); 153 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); 158 ElementsAre(constant, negate1, negate2, trace));
|
H A D | layout_assignment_test.cc | 148 auto negate1 = builder.AddInstruction( local 151 HloInstruction::CreateUnary(ashape, HloOpcode::kNegate, negate1)); 157 {negate2, negate1, add}, HloInstruction::FusionKind::kLoop);
|
Completed in 137 milliseconds