Home
last modified time | relevance | path

Searched refs:negate2 (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment_test.cc480 HloInstruction* negate2 = builder.AddInstruction( in TEST_P() local
483 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate2)); in TEST_P()
497 schedule.set_sequence(computation, {p0, p1, negate0, negate1, negate2, in TEST_P()
516 EXPECT_THAT(negate2, op::ShapeWithLayout(shape_in_alternate_mem)); in TEST_P()
689 HloInstruction* negate2 = builder.AddInstruction( in TEST_P() local
692 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate2)); in TEST_P()
716 n, o, add0, negate0, negate1, negate2, negate3, negate4, in TEST_P()
847 HloInstruction* negate2 = builder.AddInstruction( in TEST_P() local
850 HloInstruction::CreateUnary(shape, HloOpcode::kNegate, negate2)); in TEST_P()
873 computation, {p, p0, negate0, negate1, negate2, negate3, negate4, negate5, in TEST_P()
[all …]
Dhlo_computation_test.cc141 auto negate2 = builder.AddInstruction( in TEST_F() local
146 ElementsAre(constant, negate1, negate2)); in TEST_F()
158 auto negate2 = builder.AddInstruction( in TEST_F() local
164 ElementsAre(constant, negate1, negate2, trace)); in TEST_F()
Dhlo_cse_test.cc547 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in TEST_F() local
552 constant->shape(), HloOpcode::kAdd, negate2, exp2)); in TEST_F()
561 EXPECT_THAT(tuple, op::Tuple(op::Add(negate1, exp1), op::Add(negate2, exp2))); in TEST_F()
Dhlo_alias_analysis_test.cc1130 const HloInstruction* negate2 = in TEST_F() local
1134 EXPECT_EQ(analysis.GetUniqueBufferAt(negate2), in TEST_F()
Dlayout_assignment_test.cc179 auto negate2 = builder.AddInstruction( in TEST_F() local
186 {negate2, negate1, add}, HloInstruction::FusionKind::kLoop); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc488 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F() local
491 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, reverse1}, in XLA_TEST_F()
506 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F() local
509 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, broadcast1}, in XLA_TEST_F()
524 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F() local
527 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, slice1}, in XLA_TEST_F()
564 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F() local
567 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, reshape1}, in XLA_TEST_F()
582 auto negate2 = builder.AddInstruction(HloInstruction::CreateUnary( in XLA_TEST_F() local
585 ->CreateFusionInstruction(/*instructions_to_fuse=*/{negate2, transpose1}, in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_fusion_test.cc311 auto negate2 = builder.AddInstruction( in TEST_F() local
315 HloInstruction::CreateTuple({negate1, negate2, exp2})); in TEST_F()
/external/rust/crates/regex-automata/data/tests/
Dno-unicode.toml54 name = "no-unicode-negate2"