Searched refs:PropagateMemorySpace (Results 1 – 2 of 2) sorted by relevance
365 Status PropagateMemorySpace(HloModule* module);
2011 Status LayoutAssignment::PropagateMemorySpace(HloModule* module) { in PropagateMemorySpace() function in xla::LayoutAssignment2195 TF_RETURN_IF_ERROR(PropagateMemorySpace(module)); in Run()