| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_ordering_test.cc | 140 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
143 ShapeUtil::MakeShape(xla::PRED, {}), cond_param)); in TEST_F()
156 EXPECT_TRUE(ordering.ExecutesBefore(constant, cond_param)); in TEST_F()
164 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, cond_param)); in TEST_F()
166 EXPECT_FALSE(ordering.ExecutesBefore(cond_param, xla_while)); in TEST_F()
169 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, body_param)); in TEST_F()
171 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, negate)); in TEST_F()
174 EXPECT_FALSE(ordering.ExecutesBefore(body_param, cond_param)); in TEST_F()
223 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
226 ShapeUtil::MakeShape(xla::PRED, {}), cond_param)); in TEST_F()
|
| D | heap_simulator_test.cc | 49 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
52 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
54 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
91 {cond_param, cond_iter, cond_data, cond_lt}); in TEST_F()
138 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
142 ShapeUtil::MakeShape(F32, {1}), cond_param, {0}, {1}, {1})); in TEST_F()
193 std::vector<HloInstruction*> cond_vec = {cond_param, slice, reshape, zero, in TEST_F()
700 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
703 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
705 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
[all …]
|
| D | hlo_alias_analysis_test.cc | 354 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
394 GetValueDefinedAt(cond_param, {1}), in TEST_F()
405 HloPosition{cond_param, {1}}, HloPosition{negate_2, {}})); in TEST_F()
542 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
569 HloPosition{cond_param, {}})); in TEST_F()
576 HloPosition{cond_param, {0}})); in TEST_F()
583 HloPosition{body_tuple, {1}}, HloPosition{cond_param, {1}})); in TEST_F()
593 GetValueDefinedAt(cond_param, {1}), in TEST_F()
954 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
993 GetValueDefinedAt(cond_param, /*index=*/{0}), in TEST_F()
[all …]
|
| D | bfloat16_propagation_test.cc | 573 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
576 builder_cond.AddInstruction(CreateDot(shape, cond_param, cond_param)); in TEST_F()
613 EXPECT_TRUE(OutputsBF16(cond_param)); in TEST_F()
633 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
640 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {0, 0}, {1, 1}, in TEST_F()
645 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {1, 1}, {2, 2}, in TEST_F()
699 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
702 HloInstruction::CreateGetTupleElement(shape, cond_param, 0)); in TEST_F()
704 HloInstruction::CreateGetTupleElement(shape, cond_param, 1)); in TEST_F()
|
| D | hlo_dataflow_analysis_test.cc | 501 auto cond_param = cond_builder.AddInstruction( in TEST_P() local
530 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
537 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
538 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{1}).is_phi()); in TEST_P()
558 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
559 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
791 auto cond_param = cond_builder.AddInstruction( in TEST_P() local
825 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
826 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{0}).is_phi()); in TEST_P()
827 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
[all …]
|
| D | hlo_memory_scheduler_test.cc | 252 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
258 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), cond_param, in TEST_F()
|