Home
last modified time | relevance | path

Searched refs:cond_param (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering_test.cc139 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
142 ShapeUtil::MakeShape(xla::PRED, {}), cond_param)); in TEST_F()
155 EXPECT_TRUE(ordering.ExecutesBefore(constant, cond_param)); in TEST_F()
163 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, cond_param)); in TEST_F()
165 EXPECT_FALSE(ordering.ExecutesBefore(cond_param, xla_while)); in TEST_F()
168 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, body_param)); in TEST_F()
170 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, negate)); in TEST_F()
173 EXPECT_FALSE(ordering.ExecutesBefore(body_param, cond_param)); in TEST_F()
222 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
225 ShapeUtil::MakeShape(xla::PRED, {}), cond_param)); in TEST_F()
Dhlo_alias_analysis_test.cc381 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
418 &GetValueDefinedAt(cond_param, {1}), in TEST_F()
429 HloPosition{cond_param, {1}}, HloPosition{negate_2, {}})); in TEST_F()
566 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
593 HloPosition{cond_param, {}})); in TEST_F()
600 HloPosition{cond_param, {0}})); in TEST_F()
607 HloPosition{body_tuple, {1}}, HloPosition{cond_param, {1}})); in TEST_F()
615 &GetValueDefinedAt(cond_param, {1}), in TEST_F()
Dbfloat16_propagation_test.cc560 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
563 builder_cond.AddInstruction(CreateDot(shape, cond_param, cond_param)); in TEST_F()
600 EXPECT_TRUE(OutputsBF16(cond_param)); in TEST_F()
620 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
627 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {0, 0}, {1, 1}, in TEST_F()
632 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {1, 1}, {2, 2}, in TEST_F()
686 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
689 HloInstruction::CreateGetTupleElement(shape, cond_param, 0)); in TEST_F()
691 HloInstruction::CreateGetTupleElement(shape, cond_param, 1)); in TEST_F()
Dhlo_live_range_test.cc270 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
273 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
325 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_F()
Dheap_simulator_test.cc49 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()
893 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
896 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
898 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc415 auto cond_param = cond_builder.AddInstruction( in TEST_P() local
444 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
451 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
452 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{1}).is_phi()); in TEST_P()
472 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
473 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
851 auto cond_param = cond_builder.AddInstruction( in TEST_P() local
885 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
886 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{0}).is_phi()); in TEST_P()
887 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
[all …]
Dmemory_space_assignment_test.cc781 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
784 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
838 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
2590 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
2593 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
2653 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
2913 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
2916 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
2982 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
3004 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
[all …]
Dwhile_loop_concat_code_motion.cc720 auto cond_param = loop->while_condition()->parameter_instruction(0); in RewriteLoopWithConcatGroups() local
740 *cond_param->mutable_shape()->mutable_tuple_shapes(i) = in RewriteLoopWithConcatGroups()
Dbuffer_assignment_test.cc802 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
805 HloInstruction::CreateGetTupleElement(s32_, cond_param, 0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc311 auto cond_param = cond_b.AddInstruction(HloInstruction::CreateParameter( in GetFinalFftUsingCollectivePermute() local
318 HloInstruction::CreateGetTupleElement(iteration->shape(), cond_param, 4)); in GetFinalFftUsingCollectivePermute()
327 cond_param->shape(), module->AddEmbeddedComputation(cond_b.Build()), in GetFinalFftUsingCollectivePermute()
Ddot_handler.cc1754 auto cond_param = cond_b.AddInstruction(HloInstruction::CreateParameter( in PartitionBaseCase() local
1761 iteration->shape(), cond_param, 4)); in PartitionBaseCase()
1772 cond_param->shape(), module->AddEmbeddedComputation(cond_b.Build()), in PartitionBaseCase()