Выделение памяти в отдельной функции

math = (int**)malloc(m*sizeof(int));
         for (x1 = 0;x1<m;x1++)
          {
            math[x1] = (int*)malloc(n*sizeof(int));
          }

int **free_mem(int **math,int m, int n)
{
     int x1;
     math = (int**)malloc(m*sizeof(int));
         for (x1 = 0;x1<m;x1++)
          {
            math[x1] = (int*)malloc(n*sizeof(int));
          }  
          return math;
}

free_mem(math,m,n);

for (x1 = 0;x1<m;x1++)
          {
            free(math[x1]);
          }
free(math);

math = free_mem(math,m,n);

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
 
void **init_mem ( const int m, const int n, size_t pp_size, size_t p_size ) {
 
    int i = 0;
    void **tmp;
 
    if ( !( tmp = ( void** ) malloc( m * pp_size ) ) )
        exit( EXIT_FAILURE );
 
    while ( i < m ) {
        if ( !( tmp[ i ] = ( void* ) malloc ( n * p_size ) ) )
            exit( EXIT_FAILURE );
 
        ++i;
    }
 
    return tmp;
}
 
void free_mem ( int **matr, const int m, const int n ) {
 
    int i = 0;
 
    while ( i < m )
        free( matr[ i++ ] );
 
    free( matr );
}
 
void init ( int **matr, const int m, const int n ) {
 
    int i = 0, j;
 
    while ( i < m ) {
 
        j = 0;
        
        while ( j < n ){
            matr[ i ][ j ] = rand() % 10;
            ++j;
        }
 
        ++i;
    }
}
 
void print_matr ( int **matr, const int m, const int n ) {
 
    int i, j;
 
    for ( i = 0; i < m; ++i ) {
        
        for ( j = 0; j < n; ++j )
            printf( "%3i", matr[ i ][ j ] );
            
        puts("");
    }
}
 
int main ( void )
{
    const int size = 5;
    int **matr;
    
    srand( time( NULL ) );
    matr = init_mem( size, size, sizeof( int** ), sizeof( int* ) );
    init( matr, size, size );
    print_matr( matr, size, size );
    free_mem( matr, size, size );
    
    exit( EXIT_SUCCESS );
}