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.cc476 HloInstruction* negate0 = builder.AddInstruction( in TEST_P() local
479 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
497 schedule.set_sequence(computation, {p0, p1, negate0, negate1, negate2, in TEST_P()
514 EXPECT_THAT(negate0, op::ShapeWithLayout(shape_in_alternate_mem)); in TEST_P()
685 HloInstruction* negate0 = builder.AddInstruction( in TEST_P() local
688 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
716 n, o, add0, negate0, negate1, negate2, negate3, negate4, in TEST_P()
843 HloInstruction* negate0 = builder.AddInstruction( in TEST_P() local
846 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate0)); in TEST_P()
873 computation, {p, p0, negate0, negate1, negate2, negate3, negate4, negate5, in TEST_P()
[all …]
Dhlo_alias_analysis_test.cc233 auto negate0 = builder.AddInstruction( in TEST_F() local
239 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
1126 const HloInstruction* negate0 = in TEST_F() local
1136 EXPECT_NE(analysis.GetUniqueBufferAt(negate0), in TEST_F()
1171 const HloInstruction* negate0 = in TEST_F() local
1173 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.cc1721 auto negate0 = builder.AddInstruction( in TEST_F() local
1726 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
1759 auto negate0 = builder.AddInstruction( in TEST_F() local
1766 scalar_shape_, HloOpcode::kAdd, negate0, negate1)); in TEST_F()