afe.load.importers.caffe

Functions

import_caffe_model(prototxt_file_path, ...)	Loads in a caffe model from a directory path
validate_input_parameters(→ None)	Validates the user supplied input.
import_caffe_to_tvm(→ afe._tvm._defines.TVMIRModule)	Use TVM frontend to import a caffe pb model into TVM Relay IR

Module Contents

afe.load.importers.caffe.import_caffe_model(prototxt_file_path: str, caffemodel_file_path: str)

Loads in a caffe model from a directory path :param prototxt_file_path: filepath to the caffe .prototxt file :param caffemodel_file_path: filepath to the caffe .caffemodel file

afe.load.importers.caffe.validate_input_parameters(prototxt_file_path: str, caffemodel_file_path: str, shape_dict: Dict[str, Tuple[int, Ellipsis]], dtype_dict: Dict[str, afe.ir.tensor_type.ScalarType]) → None

Validates the user supplied input. :param prototxt_file_path: filepath to the caffe .prototxt file :param caffemodel_file_path: filepath to the caffe .caffemodel file :param shape_dict: dictionary of input names to input shapes (eg. (1,224,224,3)) :param dtype_dict: dictionary of input names to input types (eg. float32 or int64)

afe.load.importers.caffe.import_caffe_to_tvm(prototxt_file_path: str, caffemodel_file_path: str, shape_dict: Dict[str, Tuple[int, Ellipsis]], dtype_dict: Dict[str, afe.ir.tensor_type.ScalarType]) → afe._tvm._defines.TVMIRModule

Use TVM frontend to import a caffe pb model into TVM Relay IR :param prototxt_file_path: filepath to the caffe .prototxt file :param caffemodel_file_path: filepath to the caffe .caffemodel file :param shape_dict: dictionary of input names to input shapes (eg. (1,224,224,3)) :param dtype_dict: dictionary of input names to input types (eg. float32 or int64) :return: Imported TVM IR module.