from engines.inference_engine cimport InferenceEngine


cdef class CoreMLEngine(InferenceEngine):

    cdef object model
    cdef str input_name
    cdef tuple input_shape
    cdef list _output_names

    cdef tuple get_input_shape(self)
    cdef int get_batch_size(self)
    cdef run(self, input_data)