Home
last modified time | relevance | path

Searched refs:negate1 (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_computation_test.cc140 auto negate1 = builder.AddInstruction( in TEST_F() local
143 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
147 ElementsAre(constant, negate1, negate2)); in TEST_F()
155 auto negate1 = builder.AddInstruction( in TEST_F() local
158 builder.AddInstruction(HloInstruction::CreateTrace("foobar", negate1)); in TEST_F()
160 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
165 ElementsAre(constant, negate1, negate2, trace)); in TEST_F()
Dhlo_cse_test.cc540 auto negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F() local
545 constant->shape(), HloOpcode::kAdd, negate1, exp1)); in TEST_F()
561 EXPECT_THAT(tuple, op::Tuple(op::Add(negate1, exp1), op::Add(negate2, exp2))); in TEST_F()
Dcopy_insertion_test.cc1519 auto negate1 = builder.AddInstruction( in TEST_F() local
1521 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
1558 auto negate1 = builder.AddInstruction( in TEST_F() local
1562 scalar_shape_, HloOpcode::kAdd, negate0, negate1)); in TEST_F()
1563 builder.AddInstruction(HloInstruction::CreateTuple({add, negate1})); in TEST_F()
Dlayout_assignment_test.cc177 auto negate1 = builder.AddInstruction( in TEST_F() local
180 HloInstruction::CreateUnary(ashape, HloOpcode::kNegate, negate1)); in TEST_F()
186 {negate2, negate1, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
Dhlo_alias_analysis_test.cc235 auto negate1 = builder.AddInstruction( in TEST_F() local
239 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc306 auto negate1 = builder.AddInstruction( in TEST_F() local
315 HloInstruction::CreateTuple({negate1, negate2, exp2})); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc58 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F() local
62 ShapeUtil::MakeShape(S32, {1}), negate1, {0})); in TEST_F()