Home
last modified time | relevance | path

Searched refs:body_param (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc340 auto body_param = body_builder.AddInstruction( in TEST_F() local
343 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
345 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
393 GetValueDefinedAt(body_param, {1}), in TEST_F()
402 HloPosition{while_element_2, {}}, HloPosition{body_param, {1}}, in TEST_F()
528 auto body_param = body_builder.AddInstruction( in TEST_F() local
531 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
533 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
567 HloPosition{body_param, {}}, in TEST_F()
574 HloPosition{xla_while, {0}}, HloPosition{body_param, {0}}, in TEST_F()
[all …]
Dhlo_ordering_test.cc133 auto body_param = body_builder.AddInstruction( in TEST_F() local
136 scalar_shape, HloOpcode::kNegate, body_param)); in TEST_F()
158 EXPECT_TRUE(ordering.ExecutesBefore(constant, body_param)); in TEST_F()
163 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, body_param)); in TEST_F()
165 EXPECT_FALSE(ordering.ExecutesBefore(body_param, xla_while)); in TEST_F()
169 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, body_param)); in TEST_F()
170 EXPECT_TRUE(ordering.ExecutesBefore(convert, body_param)); in TEST_F()
174 EXPECT_FALSE(ordering.ExecutesBefore(body_param, cond_param)); in TEST_F()
216 auto body_param = body_builder.AddInstruction( in TEST_F() local
219 scalar_shape, HloOpcode::kNegate, body_param)); in TEST_F()
Dhlo_dataflow_analysis_test.cc487 auto body_param = body_builder.AddInstruction( in TEST_P() local
490 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P()
492 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P()
529 EXPECT_FALSE(analysis.ValueIsDefinedAt(body_param, /*index=*/{0})); in TEST_P()
535 EXPECT_TRUE(analysis.ValueIsDefinedAt(body_param, /*index=*/{1})); in TEST_P()
536 EXPECT_TRUE(analysis.GetValueDefinedAt(body_param, /*index=*/{1}).is_phi()); in TEST_P()
556 EXPECT_FALSE(analysis.ValueIsDefinedAt(body_param, /*index=*/{0})); in TEST_P()
557 EXPECT_FALSE(analysis.ValueIsDefinedAt(body_param, /*index=*/{1})); in TEST_P()
590 auto body_param = body_builder.AddInstruction( in TEST_P() local
593 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P()
[all …]
Dbfloat16_propagation_test.cc593 auto body_param = builder_body.AddInstruction( in TEST_F() local
596 builder_body.AddInstruction(CreateDot(shape, body_param, body_param)); in TEST_F()
612 EXPECT_TRUE(OutputsBF16(body_param)); in TEST_F()
651 auto body_param = builder_body.AddInstruction( in TEST_F() local
654 HloInstruction::CreateTranspose(shape, body_param, {0, 1})); in TEST_F()
675 EXPECT_FALSE(OutputsBF16(body_param)); in TEST_F()
726 auto body_param = builder_body.AddInstruction( in TEST_F() local
729 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_F()
731 HloInstruction::CreateGetTupleElement(shape, body_param, 1)); in TEST_F()
852 auto body_param = builder_body.AddInstruction( in TEST_F() local
[all …]
Dheap_simulator_test.cc64 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
92 schedule.set_sequence(body_computation, {body_param}); in TEST_F()
155 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
162 r1f32, HloOpcode::kSubtract, body_param, one_vector)); in TEST_F()
195 std::vector<HloInstruction*> while_body_vec = {body_param, one_vector, in TEST_F()
713 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
726 {param, while_op, body_param, cond_param, cond_iter, cond_data, cond_lt}); in TEST_F()
737 {kAlloc, tracker.BufferAt(body_param, {})}, in TEST_F()
738 {kAlloc, tracker.BufferAt(body_param, {0})}, in TEST_F()
739 {kAlloc, tracker.BufferAt(body_param, {1})}, in TEST_F()
[all …]
Dcopy_insertion_test.cc1304 auto body_param = body_builder.AddInstruction( in TEST_F() local
1307 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
1309 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
1586 auto body_param = body_builder.AddInstruction( in TEST_F() local
1589 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
1591 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
1648 auto body_param = body_builder.AddInstruction( in TEST_F() local
1651 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
1653 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
1725 auto body_param = body_builder.AddInstruction( in TEST_F() local
[all …]
Dhlo_memory_scheduler_test.cc265 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
271 r1f32, HloOpcode::kSubtract, body_param, one_vector)); in TEST_F()
Ddynamic_dimension_inference_test.cc475 auto body_param = body_builder.AddInstruction( in TEST_F() local
478 HloInstruction::CreateGetTupleElement(input_shape, body_param, 0)); in TEST_F()
480 HloInstruction::CreateGetTupleElement(input_shape, body_param, 1)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc232 absl::c_for_each(s.body_parameters, [&hashes](const BodyParam& body_param) { in AbslHashValue() argument
233 hashes.push_back(Hash64(body_param.first)); in AbslHashValue()
234 hashes.push_back(FastAttrValueHash(body_param.second)); in AbslHashValue()