Searched refs:MakeRematerializableComputation (Results 1 – 1 of 1) sorted by relevance
58 std::unique_ptr<HloComputation> MakeRematerializableComputation( in MakeRematerializableComputation() function in xla::__anon7066800c0111::HloRematerializationTest167 module->AddEntryComputation(MakeRematerializableComputation()); in TEST_F()208 module->AddEntryComputation(MakeRematerializableComputation()); in TEST_F()239 MakeRematerializableComputation(/*suffix=*/".body")); in TEST_F()275 MakeRematerializableComputation(/*suffix=*/".body")); in TEST_F()307 MakeRematerializableComputation(/*suffix=*/".inner")); in TEST_F()