Home
last modified time | relevance | path

Searched refs:negate0 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc510 HloInstruction* negate0 = builder.AddInstruction( in TEST_P() local
513 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
531 schedule.set_sequence(computation, {p0, p1, negate0, negate1, negate2, in TEST_P()
549 EXPECT_THAT(negate0, op::ShapeWithLayout(shape_in_alternate_mem)); in TEST_P()
720 HloInstruction* negate0 = builder.AddInstruction( in TEST_P() local
723 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
751 n, o, add0, negate0, negate1, negate2, negate3, negate4, in TEST_P()
878 HloInstruction* negate0 = builder.AddInstruction( in TEST_P() local
881 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
908 computation, {p, p0, negate0, negate1, negate2, negate3, negate4, negate5, in TEST_P()
[all …]
Dhlo_alias_analysis_test.cc266 auto negate0 = builder.AddInstruction( in TEST_F() local
272 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
960 const HloInstruction* negate0 = in TEST_F() local
970 EXPECT_NE(analysis.GetUniqueBufferAt(negate0), in TEST_F()
1005 const HloInstruction* negate0 = in TEST_F() local
1007 EXPECT_NE(analysis.GetUniqueBufferAt(negate0), in TEST_F()
Dheap_simulator_test.cc539 auto negate0 = builder.AddInstruction( in TEST_F() local
545 negate0, negate1)); in TEST_F()
550 {paramA, negate, fusion, element0, element1, negate0, negate1}, {}, in TEST_F()
560 {kAlloc, tracker.BufferAt(negate0, {})}, in TEST_F()
562 {kFree, tracker.BufferAt(negate0, {})}, in TEST_F()
Dcopy_insertion_test.cc1567 auto negate0 = builder.AddInstruction( in TEST_F() local
1572 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
1605 auto negate0 = builder.AddInstruction( in TEST_F() local
1612 scalar_shape_, HloOpcode::kAdd, negate0, negate1)); in TEST_F()