Publishing R5 content (#72)
[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 public:
37         _requests, inputs, outputs
38
39 cdef class IEPlugin:
40     cdef C.IEPlugin impl
41     cpdef ExecutableNetwork load(self, IENetwork network, int num_requests = ?, config = ?)
42     cpdef void set_config(self, config)
43     cpdef void add_cpu_extension(self, str extension_path) except *
44     cpdef void set_initial_affinity(self, IENetwork network) except *
45     cpdef set get_supported_layers(self, IENetwork net)
46
47 cdef class IENetLayer:
48     cdef C.IENetLayer impl
49
50 cdef class InputInfo:
51     cdef C.InputInfo impl
52
53 cdef class OutputInfo:
54     cdef C.OutputInfo impl
55
56 cdef class LayersStatsMap(dict):
57     cdef C.IENetwork net_impl