Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc478 HloInstruction* negate1 = builder.AddInstruction( in TEST_P() local
481 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate1)); in TEST_P()
497 schedule.set_sequence(computation, {p0, p1, negate0, negate1, negate2, in TEST_P()
515 EXPECT_THAT(negate1, op::ShapeWithLayout(shape_in_alternate_mem)); in TEST_P()
687 HloInstruction* negate1 = builder.AddInstruction( in TEST_P() local
690 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate1)); in TEST_P()
716 n, o, add0, negate0, negate1, negate2, negate3, negate4, in TEST_P()
845 HloInstruction* negate1 = builder.AddInstruction( in TEST_P() local
848 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate1)); in TEST_P()
873 computation, {p, p0, negate0, negate1, negate2, negate3, negate4, negate5, in TEST_P()
[all …]
Dhlo_computation_test.cc139 auto negate1 = builder.AddInstruction( in TEST_F() local
142 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
146 ElementsAre(constant, negate1, negate2)); in TEST_F()
154 auto negate1 = builder.AddInstruction( in TEST_F() local
157 builder.AddInstruction(HloInstruction::CreateTrace("foobar", negate1)); in TEST_F()
159 HloInstruction::CreateUnary(r0f32_, HloOpcode::kNegate, negate1)); in TEST_F()
164 ElementsAre(constant, negate1, negate2, trace)); in TEST_F()
Dheap_simulator_test.cc541 auto negate1 = builder.AddInstruction( in TEST_F() local
545 negate0, negate1)); in TEST_F()
550 {paramA, negate, fusion, element0, element1, negate0, negate1}, {}, in TEST_F()
563 {kAlloc, tracker.BufferAt(negate1, {})}, in TEST_F()
565 {kFree, tracker.BufferAt(negate1, {})}, in TEST_F()
610 auto negate1 = builder.AddInstruction( in TEST_F() local
615 {paramA, negate, fusion, element1, negate1}, {}, in TEST_F()
626 {kAlloc, tracker.BufferAt(negate1, {})}, in TEST_F()
629 {kFree, tracker.BufferAt(negate1, {})}, 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()
Dhlo_alias_analysis_test.cc235 auto negate1 = builder.AddInstruction( in TEST_F() local
239 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
1128 const HloInstruction* negate1 = in TEST_F() local
1132 EXPECT_EQ(analysis.GetUniqueBufferAt(negate1), in TEST_F()
Dcopy_insertion_test.cc1724 auto negate1 = builder.AddInstruction( in TEST_F() local
1726 builder.AddInstruction(HloInstruction::CreateTuple({negate0, negate1})); in TEST_F()
1762 auto negate1 = builder.AddInstruction( in TEST_F() local
1766 scalar_shape_, HloOpcode::kAdd, negate0, negate1)); in TEST_F()
1767 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()
/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/rust/crates/regex-automata/data/tests/
Dno-unicode.toml47 name = "no-unicode-negate1"
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dinstruction_fusion_test.cc59 HloInstruction* negate1 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F() local
63 ShapeUtil::MakeShape(S32, {1}), negate1, {})); in TEST_F()