File tree Expand file tree Collapse file tree 6 files changed +9
-9
lines changed Expand file tree Collapse file tree 6 files changed +9
-9
lines changed Original file line number Diff line number Diff line change @@ -29,7 +29,7 @@ impl Model {
29
29
30
30
/// Get the pointer to the underlying [`ov_model_t`].
31
31
#[ inline]
32
- pub ( crate ) fn as_ptr ( & self ) -> * mut ov_model_t {
32
+ pub ( crate ) fn as_ptr ( & self ) -> * const ov_model_t {
33
33
self . ptr
34
34
}
35
35
@@ -55,7 +55,7 @@ impl Model {
55
55
index,
56
56
std:: ptr:: addr_of_mut!( node)
57
57
) ) ?;
58
- Ok ( Node :: new ( node) )
58
+ Ok ( Node :: from_ptr ( node) )
59
59
}
60
60
61
61
/// Retrieve the output node by index.
@@ -66,7 +66,7 @@ impl Model {
66
66
index,
67
67
std:: ptr:: addr_of_mut!( node)
68
68
) ) ?;
69
- Ok ( Node :: new ( node) )
69
+ Ok ( Node :: from_ptr ( node) )
70
70
}
71
71
72
72
/// Retrieve the constant output node by index.
@@ -77,7 +77,7 @@ impl Model {
77
77
index,
78
78
std:: ptr:: addr_of_mut!( node)
79
79
) ) ?;
80
- Ok ( Node :: new ( node) )
80
+ Ok ( Node :: from_ptr ( node) )
81
81
}
82
82
83
83
/// Returns `true` if the model contains dynamic shapes.
Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ pub struct Node {
14
14
impl Node {
15
15
/// Create a new [`Port`] from [`ov_output_const_port_t`].
16
16
#[ inline]
17
- pub ( crate ) fn new ( ptr : * mut ov_output_const_port_t ) -> Self {
17
+ pub ( crate ) fn from_ptr ( ptr : * mut ov_output_const_port_t ) -> Self {
18
18
Self { ptr }
19
19
}
20
20
Original file line number Diff line number Diff line change @@ -26,7 +26,7 @@ impl InferRequest {
26
26
}
27
27
28
28
/// Assign a [`Tensor`] to the input on the model.
29
- pub fn set_tensor ( & mut self , name : & str , tensor : Tensor ) -> Result < ( ) > {
29
+ pub fn set_tensor ( & mut self , name : & str , tensor : & Tensor ) -> Result < ( ) > {
30
30
try_unsafe ! ( ov_infer_request_set_tensor(
31
31
self . ptr,
32
32
cstr!( name) ,
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ fn classify_alexnet() -> anyhow::Result<()> {
49
49
// Compile the model and infer the results.
50
50
let mut executable_model = core. compile_model ( & new_model, DeviceType :: CPU ) ?;
51
51
let mut infer_request = executable_model. create_infer_request ( ) ?;
52
- infer_request. set_tensor ( "data" , tensor) ?;
52
+ infer_request. set_tensor ( "data" , & tensor) ?;
53
53
infer_request. infer ( ) ?;
54
54
let mut results = infer_request. get_tensor ( & output_port. get_name ( ) ?) ?;
55
55
Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ fn classify_inception() -> anyhow::Result<()> {
46
46
// Compile the model and infer the results.
47
47
let mut executable_model = core. compile_model ( & new_model, DeviceType :: CPU ) ?;
48
48
let mut infer_request = executable_model. create_infer_request ( ) ?;
49
- infer_request. set_tensor ( "input" , tensor) ?;
49
+ infer_request. set_tensor ( "input" , & tensor) ?;
50
50
infer_request. infer ( ) ?;
51
51
let mut results = infer_request. get_tensor ( & output_port. get_name ( ) ?) ?;
52
52
Original file line number Diff line number Diff line change @@ -49,7 +49,7 @@ fn classify_mobilenet() -> anyhow::Result<()> {
49
49
// Compile the model and infer the results.
50
50
let mut executable_model = core. compile_model ( & new_model, DeviceType :: CPU ) ?;
51
51
let mut infer_request = executable_model. create_infer_request ( ) ?;
52
- infer_request. set_tensor ( "input" , tensor) ?;
52
+ infer_request. set_tensor ( "input" , & tensor) ?;
53
53
infer_request. infer ( ) ?;
54
54
let mut results = infer_request. get_tensor ( & output_port. get_name ( ) ?) ?;
55
55
You can’t perform that action at this time.
0 commit comments