     layout->set_memory_space(memory_space_assignment->alternate_memory_space_);   }   TF_RETURN_IF_ERROR(PropagateMemorySpaceToBitcasts(*memory_space_assignment));   return Status::OK(); }  Status MemorySpaceAssignment::