Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add get_[inputs|outputs]_len to CNNNetwork #34

Merged
merged 2 commits into from
Mar 24, 2022
Merged

Add get_[inputs|outputs]_len to CNNNetwork #34

merged 2 commits into from
Mar 24, 2022

Conversation

abrown
Copy link
Contributor

@abrown abrown commented Mar 24, 2022

This change is an adaptation of #33 that allows for the same kind of functionality. It also adds tests checking the newly-added functions. With this in place, users will be able to set input layouts with something like:

for i in 0..network.get_inputs_len() {
  let name = network.get_input_name(i)?;
  network.set_input_layout(&name, Layout::NHWC)?;
}

brianjjones and others added 2 commits March 23, 2022 17:51
This function will initialize the layout of all
inputs to the layout provided.
This change adds tests verifying that the library can retrieve the
number of inputs and outputs.
@abrown abrown merged commit d0330e6 into main Mar 24, 2022
@abrown abrown deleted the fix_layout branch March 24, 2022 15:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants