Searched defs:body_computation (Results 1 – 8 of 8) sorted by relevance
71 auto body_computation = MakeRematerializableComputation(); in TEST_F() local
158 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() local194 HloComputation* body_computation = module->AddEmbeddedComputation( in TEST_F() local
308 HloComputation* body_computation = in TEST_F() local
283 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() local
67 HloComputation* body_computation = in TEST_F() local164 auto body_computation = module->AddEmbeddedComputation(body_builder.Build()); in TEST_F() local908 HloComputation* body_computation = in TEST_F() local
786 HloComputation* body_computation = in TEST_P() local2669 HloComputation* body_computation = in TEST_P() local2980 HloComputation* body_computation = in TEST_P() local3085 HloComputation* body_computation = in TEST_P() local3719 HloComputation* body_computation; in TEST_P() local
815 HloComputation* body_computation = in TEST_F() local1025 auto body_computation = in TEST_F() local
2744 HloComputation* body_computation = in TEST_F() local