afe.load.importers.keras
========================

.. py:module:: afe.load.importers.keras


Functions
---------

.. autoapisummary::

   afe.load.importers.keras.import_keras_model
   afe.load.importers.keras.validate_input_parameters
   afe.load.importers.keras.import_keras_to_tvm


Module Contents
---------------

.. py:function:: import_keras_model(file_path: str, input_shapes: List[Tuple[int, Ellipsis]])

   Load a Keras model from a file path and a list of input shapes
   :param file_path: str. Filepath to the keras .h5 file
   :param input_shapes: List[Tupe[int, ...]]. List of input shapes (eg. (1,224,224,3))
   return: A Keras model


.. py:function:: validate_input_parameters(keras_file_path: str, shape_dict: Dict[str, Tuple[int, Ellipsis]]) -> None

   Validates the user supplied input.
   :param keras_file_path: filepath to the keras .h5 file
   :param shape_dict: dictionary of input names to input shapes (eg. (1,224,224,3))


.. py:function:: import_keras_to_tvm(keras_file_path: str, shape_dict: Dict[str, Tuple[int, Ellipsis]], layout: str) -> afe._tvm._defines.TVMIRModule

   :param keras_file_path: filepath to the keras .h5 file
   :param shape_dict: dictionary of input names to input shapes (eg. (1,224,224,3))
   :param layout: How data should be arranged in the imported model.
   :return: Imported TVM IR module.