Searched refs:AssignMemorySpace (Results 1 – 1 of 1) sorted by relevance
65 return AssignMemorySpace( in AssignMemorySpaceUsingCostAnalysis()72 std::unique_ptr<PresetAssignments> AssignMemorySpace( in AssignMemorySpace() function in xla::__anona324b8620111::MemorySpaceAssignmentTest78 return AssignMemorySpace(module, max_outstanding_async_copies, in AssignMemorySpace()83 std::unique_ptr<PresetAssignments> AssignMemorySpace( in AssignMemorySpace() function in xla::__anona324b8620111::MemorySpaceAssignmentTest416 AssignMemorySpace(module.get()); in TEST_P()445 auto preset_assignments = AssignMemorySpace(module.get()); in TEST_P()501 AssignMemorySpace(module.get()); in TEST_P()533 AssignMemorySpace(module.get()); in TEST_P()546 AssignMemorySpace(module.get(), /*max_outstanding_async_copies=*/0); in TEST_P()555 AssignMemorySpace(module.get(), /*max_outstanding_async_copies=*/1); in TEST_P()[all …]