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.cc334 auto body_param = body_builder.AddInstruction( in TEST_F() local
520 auto body_param = body_builder.AddInstruction( in TEST_F() local
615 auto body_param = body_builder.AddInstruction( in TEST_F() local
786 auto body_param = body_builder.AddInstruction( in TEST_F() local
936 auto body_param = body_builder.AddInstruction( in TEST_F() local
1213 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.cc623 auto body_param = builder_body.AddInstruction( in TEST_F() local
681 auto body_param = builder_body.AddInstruction( in TEST_F() local
756 auto body_param = builder_body.AddInstruction( in TEST_F() local
882 auto body_param = builder_body.AddInstruction( in TEST_F() local
Dhlo_dataflow_analysis_test.cc394 auto body_param = body_builder.AddInstruction( in TEST_P() local
497 auto body_param = body_builder.AddInstruction( in TEST_P() local
780 auto body_param = body_builder.AddInstruction( in TEST_P() local
833 auto body_param = body_builder.AddInstruction( in TEST_P() local
1080 auto body_param = body_builder.AddInstruction( in TEST_P() local
1440 auto body_param = body_builder.AddInstruction( in TEST_P() local
Dhlo_live_range_test.cc283 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
Dheap_simulator_test.cc65 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
156 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
906 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
Dcopy_insertion_test.cc1514 auto body_param = body_builder.AddInstruction( in TEST_F() local
1789 auto body_param = body_builder.AddInstruction( in TEST_F() local
1851 auto body_param = body_builder.AddInstruction( in TEST_F() local
1928 auto body_param = body_builder.AddInstruction( in TEST_F() local
Dmemory_space_assignment_test.cc761 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
1449 const HloInstruction* body_param = in TEST_P() local
2644 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
2965 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
3056 HloInstruction* body_param = body_builder.AddInstruction( in TEST_P() local
Ddynamic_dimension_inference_test.cc743 auto body_param = body_builder.AddInstruction( in TEST_F() local
Dbuffer_assignment_test.cc796 HloInstruction* body_param = body_builder.AddInstruction( in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2948 auto body_param = body->parameter_instruction(0); in SinkInputNodesIntoWindowedDotGeneralLoopOnContractingDimensions() local
3179 auto body_param = body->parameter_instruction(0); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc215 absl::c_for_each(s.body_parameters, [&hashes](const BodyParam& body_param) { in AbslHashValue()