Searched refs:CheckRootInDefaultMemory (Results 1 – 1 of 1) sorted by relevance
176 CheckRootInDefaultMemory(module); in AssignMemorySpace()218 void CheckRootInDefaultMemory(const HloModule* module) { in CheckRootInDefaultMemory() function in xla::__anon150a2bd80111::MemorySpaceAssignmentTest