Skip to content

Commit

Permalink
Update protos and ops
Browse files Browse the repository at this point in the history
  • Loading branch information
adamcrume committed Feb 12, 2023
1 parent f6cb5bc commit 0557f8b
Show file tree
Hide file tree
Showing 10 changed files with 9,240 additions and 553 deletions.
3,291 changes: 3,291 additions & 0 deletions src/eager/op/raw_ops.rs

Large diffs are not rendered by default.

10 changes: 2 additions & 8 deletions src/graph.rs
Original file line number Diff line number Diff line change
Expand Up @@ -167,10 +167,7 @@ impl ImportGraphDefOptions {
/// unique. Defaults to false.
pub fn set_uniquify_names(&mut self, uniquify_names: bool) {
unsafe {
tf::TF_ImportGraphDefOptionsSetUniquifyNames(
self.inner,
u8::from(uniquify_names),
);
tf::TF_ImportGraphDefOptionsSetUniquifyNames(self.inner, u8::from(uniquify_names));
}
}

Expand All @@ -179,10 +176,7 @@ impl ImportGraphDefOptions {
/// treated as an error. This option has no effect if no prefix is specified.
pub fn set_uniquify_prefix(&mut self, uniquify_prefix: bool) {
unsafe {
tf::TF_ImportGraphDefOptionsSetUniquifyPrefix(
self.inner,
u8::from(uniquify_prefix),
);
tf::TF_ImportGraphDefOptionsSetUniquifyPrefix(self.inner, u8::from(uniquify_prefix));
}
}

Expand Down
Loading

0 comments on commit 0557f8b

Please sign in to comment.