from engines.inference_engine cimport InferenceEngine


cdef class OnnxEngine(InferenceEngine):

    cdef public object session
    cdef object _cpu_session
    cdef object model_inputs
    cdef str input_name
    cdef object input_shape
    cdef object _resolved_input_hw

    cdef tuple _resolve_input_hw(self, object metadata)
    cdef tuple get_input_shape(self)
    cdef run(self, input_data)
