parameters: (self, matmul_fn)
return_statement: return matmul_fn
docstring: Perform the Matmul registration.
  
      Args:
        matmul_fn: The function to use for the Matmul.
  
      Returns:
        matmul_fn
  
      Raises:
        TypeError: if matmul_fn is not a callable.
        ValueError: if a Matmul function has already been registered for
          the given argument classes.
docstring_summary: Perform the Matmul registration.
docstring_tokens: Perform the Matmul registration .
function: def __call__(self, matmul_fn):
      """Perform the Matmul registration.
  
      Args:
        matmul_fn: The function to use for the Matmul.
  
      Returns:
        matmul_fn
  
      Raises:
        TypeError: if matmul_fn is not a callable.
        ValueError: if a Matmul function has already been registered for
          the given argument classes.
      """
      if not callable(matmul_fn):
        raise TypeError(
            "matmul_fn must be callable, received: {}".format(matmul_fn))
      if self._key in _MATMUL:
        raise ValueError("Matmul({}, {}) has already been registered.".format(
            self._key[0].__name__,
            self._key[1].__name__))
      _MATMUL[self._key] = matmul_fn
      return matmul_fn
function_tokens: def __call__ ( self , matmul_fn ) : if not callable ( matmul_fn ) : raise TypeError ( "matmul_fn must be callable, received: {}" . format ( matmul_fn ) ) if self . _key in _MATMUL : raise ValueError ( "Matmul({}, {}) has already been registered." . format ( self . _key [ 0 ] . __name__ , self . _key [ 1 ] . __name__ ) ) _MATMUL [ self . _key ] = matmul_fn return matmul_fn