Home
last modified time | relevance | path

Searched refs:cond_param (Results 1 – 10 of 10) 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.cc348 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
386 GetValueDefinedAt(cond_param, {1}), in TEST_F()
397 HloPosition{cond_param, {1}}, HloPosition{negate_2, {}})); in TEST_F()
534 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
561 HloPosition{cond_param, {}})); in TEST_F()
568 HloPosition{cond_param, {0}})); in TEST_F()
575 HloPosition{body_tuple, {1}}, HloPosition{cond_param, {1}})); in TEST_F()
585 GetValueDefinedAt(cond_param, {1}), in TEST_F()
946 auto cond_param = cond_builder.AddInstruction( in TEST_F() local
985 GetValueDefinedAt(cond_param, /*index=*/{0}), in TEST_F()
[all …]
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()
Dbfloat16_propagation_test.cc603 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
606 builder_cond.AddInstruction(CreateDot(shape, cond_param, cond_param)); in TEST_F()
643 EXPECT_TRUE(OutputsBF16(cond_param)); in TEST_F()
663 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
670 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {0, 0}, {1, 1}, in TEST_F()
675 ShapeUtil::MakeShape(F32, {1, 1}), cond_param, {1, 1}, {2, 2}, in TEST_F()
729 auto cond_param = builder_cond.AddInstruction( in TEST_F() local
732 HloInstruction::CreateGetTupleElement(shape, cond_param, 0)); in TEST_F()
734 HloInstruction::CreateGetTupleElement(shape, cond_param, 1)); in TEST_F()
Dheap_simulator_test.cc50 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
53 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 0)); in TEST_F()
55 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_F()
92 {cond_param, cond_iter, cond_data, cond_lt}); in TEST_F()
139 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
143 ShapeUtil::MakeShape(F32, {1}), cond_param, {0}, {1}, {1})); in TEST_F()
194 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.cc408 auto cond_param = cond_builder.AddInstruction( in TEST_P() local
437 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
444 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
445 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{1}).is_phi()); in TEST_P()
465 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
466 EXPECT_FALSE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
844 auto cond_param = cond_builder.AddInstruction( in TEST_P() local
878 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{0})); in TEST_P()
879 EXPECT_TRUE(analysis.GetValueDefinedAt(cond_param, /*index=*/{0}).is_phi()); in TEST_P()
880 EXPECT_TRUE(analysis.ValueIsDefinedAt(cond_param, /*index=*/{1})); in TEST_P()
[all …]
Dmemory_space_assignment_test.cc746 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
749 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
803 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
2629 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
2632 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
2692 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
2952 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
2955 HloInstruction::CreateGetTupleElement(scalar_shape, cond_param, 1)); in TEST_P()
3021 {cond_param, cond_iter, cond_limit, cond_lt}); in TEST_P()
3043 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_P() local
[all …]
Dbuffer_assignment_test.cc783 HloInstruction* cond_param = cond_builder.AddInstruction( in TEST_F() local
786 HloInstruction::CreateGetTupleElement(s32_, cond_param, 0)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc313 auto cond_param = cond_b.AddInstruction(HloInstruction::CreateParameter( in GetFinalFftUsingCollectivePermute() local
320 HloInstruction::CreateGetTupleElement(iteration->shape(), cond_param, 4)); in GetFinalFftUsingCollectivePermute()
329 cond_param->shape(), module->AddEmbeddedComputation(cond_b.Build()), in GetFinalFftUsingCollectivePermute()
Ddot_handler.cc1402 auto cond_param = cond_b.AddInstruction(HloInstruction::CreateParameter( in PartitionBaseCase() local
1409 iteration->shape(), cond_param, 4)); in PartitionBaseCase()
1420 cond_param->shape(), module->AddEmbeddedComputation(cond_b.Build()), in PartitionBaseCase()