mirror of
https://github.com/azaion/annotations.git
synced 2026-04-22 22:26:31 +00:00
Alex Bezdieniezhnykh
25 lines
456 B
Cython
25 lines
456 B
Cython
from inference_engine cimport InferenceEngine
|
|
|
|
|
|
cdef class TensorRTEngine(InferenceEngine):
|
|
|
|
cdef public object context
|
|
|
|
cdef public object d_input
|
|
cdef public object d_output
|
|
cdef str input_name
|
|
cdef object input_shape
|
|
|
|
cdef object h_output
|
|
cdef str output_name
|
|
cdef object output_shape
|
|
|
|
cdef object stream
|
|
|
|
|
|
cpdef tuple get_input_shape(self)
|
|
|
|
cpdef int get_batch_size(self)
|
|
|
|
cpdef run(self, input_data)
|