Publishing 2019 R1 content
[platform/upstream/dldt.git] / inference-engine / ie_bridges / python / src / openvino / inference_engine / ie_api.pxd
1 from .cimport ie_api_impl_defs as C
2 from .ie_api_impl_defs cimport Blob, TensorDesc
3
4 from libcpp.string cimport string
5 from libcpp.vector cimport vector
6 from libcpp.memory cimport unique_ptr
7
8 cdef class BlobBuffer:
9     cdef Blob.Ptr ptr
10     cdef char*format
11     cdef vector[Py_ssize_t] shape
12     cdef vector[Py_ssize_t] strides
13     cdef reset(self, Blob.Ptr &)
14     cdef char*_get_blob_format(self, const TensorDesc & desc)
15
16     cdef public:
17         total_stride, item_size
18
19 cdef class InferRequest:
20     cdef C.InferRequestWrap *impl
21
22     cpdef BlobBuffer _get_blob_buffer(self, const string & blob_name)
23
24     cpdef infer(self, inputs = ?)
25     cpdef async_infer(self, inputs = ?)
26     cpdef wait(self, timeout = ?)
27     cpdef get_perf_counts(self)
28     cdef public:
29         _inputs_list, _outputs_list
30
31 cdef class IENetwork:
32     cdef C.IENetwork impl
33
34 cdef class ExecutableNetwork:
35     cdef unique_ptr[C.IEExecNetwork] impl
36     cdef C.IEPlugin plugin_impl
37     cdef public:
38         _requests, inputs, outputs
39
40 cdef class IEPlugin:
41     cdef C.IEPlugin impl
42     cpdef ExecutableNetwork load(self, IENetwork network, int num_requests = ?, config = ?)
43     cpdef void set_config(self, config)
44     cpdef void add_cpu_extension(self, str extension_path) except *
45     cpdef void set_initial_affinity(self, IENetwork network) except *
46     cpdef set get_supported_layers(self, IENetwork net)
47
48 cdef class IENetLayer:
49     cdef C.IENetLayer impl
50
51 cdef class InputInfo:
52     cdef C.InputInfo impl
53
54 cdef class OutputInfo:
55     cdef C.OutputInfo impl
56
57 cdef class LayersStatsMap(dict):
58     cdef C.IENetwork net_impl