Searched defs:negate1 (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dinstruction_fusion_test.cc53 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( local
57 ShapeUtil::MakeShape(S32, {1}), negate1, {0}));
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_cse_test.cc338 // 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 Dhlo_computation_test.cc133 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 Dlayout_assignment_test.cc148 auto negate1 = builder.AddInstruction( local
151 HloInstruction::CreateUnary(ashape, HloOpcode::kNegate, negate1));
157 {negate2, negate1, add}, HloInstruction::FusionKind::kLoop);

Completed in 137 milliseconds