Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc367 auto body_param = body_builder.AddInstruction( in TEST_F() local
552 auto body_param = body_builder.AddInstruction( in TEST_F() local
645 auto body_param = body_builder.AddInstruction( in TEST_F() local
816 auto body_param = body_builder.AddInstruction( in TEST_F() local
Dhlo_ordering_test.cc132 auto body_param = body_builder.AddInstruction( in TEST_F() local
215 auto body_param = body_builder.AddInstruction( in TEST_F() local
Dbfloat16_propagation_test.cc580 auto body_param = builder_body.AddInstruction( in TEST_F() local
638 auto body_param = builder_body.AddInstruction( in TEST_F() local
713 auto body_param = builder_body.AddInstruction( in TEST_F() local
839 auto body_param = builder_body.AddInstruction( in TEST_F() local
Dhlo_live_range_test.cc283 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
Dheap_simulator_test.cc64 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
155 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
906 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
Dhlo_dataflow_analysis_test.cc401 auto body_param = body_builder.AddInstruction( in TEST_P() local
504 auto body_param = body_builder.AddInstruction( in TEST_P() local
787 auto body_param = body_builder.AddInstruction( in TEST_P() local
840 auto body_param = body_builder.AddInstruction( in TEST_P() local
1367 auto body_param = body_builder.AddInstruction( in TEST_P() local
Dcopy_insertion_test.cc1360 auto body_param = body_builder.AddInstruction( in TEST_F() local
1635 auto body_param = body_builder.AddInstruction( in TEST_F() local
1697 auto body_param = body_builder.AddInstruction( in TEST_F() local
1774 auto body_param = body_builder.AddInstruction( in TEST_F() local
Dmemory_space_assignment_test.cc796 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
1456 const HloInstruction* body_param = in TEST_P() local
2605 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
2926 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
3017 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
Ddynamic_dimension_inference_test.cc750 auto body_param = body_builder.AddInstruction( in TEST_F() local
Dbuffer_assignment_test.cc815 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc218 absl::c_for_each(s.body_parameters, [&hashes](const BodyParam& body_param) { in AbslHashValue()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc3884 auto body_param = body->parameter_instruction(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
4158 auto* body_param = body->parameter_instruction(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local