Skip to content

Commit 73ed289

Browse files
authored
Merge pull request tensorflow#88 from adamcrume/master
Fix memory issues uncovered by valgrind
2 parents 27bc757 + 769258b commit 73ed289

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/graph.rs

+2-2
Original file line numberDiff line numberDiff line change
@@ -959,7 +959,7 @@ impl<'a> OperationDescription<'a> {
959959
unsafe {
960960
tf::TF_SetAttrTensor(self.inner,
961961
c_attr_name.as_ptr(),
962-
value.into_ptr(),
962+
value.inner,
963963
status.inner());
964964
}
965965
status.into_result()
@@ -973,7 +973,7 @@ impl<'a> OperationDescription<'a> {
973973
let c_attr_name = try!(CString::new(attr_name));
974974
let mut status = Status::new();
975975
unsafe {
976-
let ptrs: Vec<*mut tf::TF_Tensor> = value.into_iter().map(|x| x.into_ptr()).collect();
976+
let ptrs: Vec<*mut tf::TF_Tensor> = value.into_iter().map(|x| x.inner).collect();
977977
tf::TF_SetAttrTensorList(self.inner,
978978
c_attr_name.as_ptr(),
979979
ptrs.as_ptr() as *const *const tf::TF_Tensor,

0 commit comments

Comments
 (0)