Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc334 auto body_param = body_builder.AddInstruction( in TEST_F() local
337 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
339 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
385 GetValueDefinedAt(body_param, {1}), in TEST_F()
394 HloPosition{while_element_2, {}}, HloPosition{body_param, {1}}, in TEST_F()
520 auto body_param = body_builder.AddInstruction( in TEST_F() local
523 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
525 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
559 HloPosition{body_param, {}}, in TEST_F()
566 HloPosition{xla_while, {0}}, HloPosition{body_param, {0}}, in TEST_F()
[all …]
Dhlo_ordering_test.cc132 auto body_param = body_builder.AddInstruction( in TEST_F() local
135 scalar_shape, HloOpcode::kNegate, body_param)); in TEST_F()
157 EXPECT_TRUE(ordering.ExecutesBefore(constant, body_param)); in TEST_F()
162 EXPECT_FALSE(ordering.ExecutesBefore(xla_while, body_param)); in TEST_F()
164 EXPECT_FALSE(ordering.ExecutesBefore(body_param, xla_while)); in TEST_F()
168 EXPECT_TRUE(ordering.ExecutesBefore(cond_param, body_param)); in TEST_F()
169 EXPECT_TRUE(ordering.ExecutesBefore(convert, body_param)); in TEST_F()
173 EXPECT_FALSE(ordering.ExecutesBefore(body_param, cond_param)); in TEST_F()
215 auto body_param = body_builder.AddInstruction( in TEST_F() local
218 scalar_shape, HloOpcode::kNegate, body_param)); in TEST_F()
Dhlo_dataflow_analysis_test.cc394 auto body_param = body_builder.AddInstruction( in TEST_P() local
397 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P()
399 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_P()
436 EXPECT_FALSE(analysis.ValueIsDefinedAt(body_param, /*index=*/{0})); in TEST_P()
442 EXPECT_TRUE(analysis.ValueIsDefinedAt(body_param, /*index=*/{1})); in TEST_P()
443 EXPECT_TRUE(analysis.GetValueDefinedAt(body_param, /*index=*/{1}).is_phi()); in TEST_P()
463 EXPECT_FALSE(analysis.ValueIsDefinedAt(body_param, /*index=*/{0})); in TEST_P()
464 EXPECT_FALSE(analysis.ValueIsDefinedAt(body_param, /*index=*/{1})); in TEST_P()
497 auto body_param = body_builder.AddInstruction( in TEST_P() local
500 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_P()
[all …]
Dhlo_live_range_test.cc283 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
286 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_F()
288 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_F()
327 {body_param, body_iter, body_data, body_iter_increment, in TEST_F()
Dbfloat16_propagation_test.cc623 auto body_param = builder_body.AddInstruction( in TEST_F() local
626 builder_body.AddInstruction(CreateDot(shape, body_param, body_param)); in TEST_F()
642 EXPECT_TRUE(OutputsBF16(body_param)); in TEST_F()
681 auto body_param = builder_body.AddInstruction( in TEST_F() local
684 HloInstruction::CreateTranspose(shape, body_param, {0, 1})); in TEST_F()
705 EXPECT_FALSE(OutputsBF16(body_param)); in TEST_F()
756 auto body_param = builder_body.AddInstruction( in TEST_F() local
759 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_F()
761 HloInstruction::CreateGetTupleElement(shape, body_param, 1)); in TEST_F()
882 auto body_param = builder_body.AddInstruction( in TEST_F() local
[all …]
Dcopy_insertion_test.cc1514 auto body_param = body_builder.AddInstruction( in TEST_F() local
1517 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
1519 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
1789 auto body_param = body_builder.AddInstruction( in TEST_F() local
1792 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
1794 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
1851 auto body_param = body_builder.AddInstruction( in TEST_F() local
1854 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 0)); in TEST_F()
1856 HloInstruction::CreateGetTupleElement(scalar_shape_, body_param, 1)); in TEST_F()
1928 auto body_param = body_builder.AddInstruction( in TEST_F() local
[all …]
Dheap_simulator_test.cc65 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
93 schedule.set_sequence(body_computation, {body_param}); in TEST_F()
156 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
163 r1f32, HloOpcode::kSubtract, body_param, one_vector)); in TEST_F()
196 std::vector<HloInstruction*> while_body_vec = {body_param, one_vector, in TEST_F()
906 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
919 {param, while_op, body_param, cond_param, cond_iter, cond_data, cond_lt}); in TEST_F()
Dmemory_space_assignment_test.cc761 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
764 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_P()
766 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_P()
805 {body_param, body_iter, body_data, body_iter_increment, in TEST_P()
1449 const HloInstruction* body_param = in TEST_P() local
1452 for (const HloInstruction* user : body_param->users()) { in TEST_P()
2644 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
2647 HloInstruction::CreateGetTupleElement(scalar_shape, body_param, 1)); in TEST_P()
2649 HloInstruction::CreateGetTupleElement(shape, body_param, 0)); in TEST_P()
2694 {body_param, body_iter, body_data, body_iter_increment, in TEST_P()
[all …]
Ddynamic_dimension_inference_test.cc743 auto body_param = body_builder.AddInstruction( in TEST_F() local
746 HloInstruction::CreateGetTupleElement(input_shape, body_param, 0)); in TEST_F()
748 HloInstruction::CreateGetTupleElement(input_shape, body_param, 1)); in TEST_F()
Dbuffer_assignment_test.cc796 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
799 HloInstruction::CreateGetTupleElement(s32_, body_param, 0)); in TEST_F()
801 HloInstruction::CreateGetTupleElement(f32vec10_color1, body_param, 1)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc215 absl::c_for_each(s.body_parameters, [&hashes](const BodyParam& body_param) { in AbslHashValue() argument
216 hashes.push_back(Hash64(body_param.first)); in AbslHashValue()
217 hashes.push_back(FastAttrValueHash(body_param.second)); in AbslHashValue()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2948 auto body_param = body->parameter_instruction(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
2949 auto old_body_param_users = body_param->users(); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2953 body_param, body->root_instruction()}) { in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
2961 new_input_subtuple->shape(), body_param, non_windowed_operand_index)); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions()
3179 auto body_param = body->parameter_instruction(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
3185 body_param, body_root}) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3191 new_input_subtuple->shape(), body_param, 2)); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()