Skip to content

Add graph walking functions to Graph and GraphOperation #232

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.

Sign up for GitHub

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

Merged
merged 8 commits into from
Mar 18, 2021
Merged

Add graph walking functions to Graph and GraphOperation #232

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
2c7fa1b
Add graph walking functions to Graph and GraphOperation
rnett Mar 4, 2021
b3306d5
Add subgraph option to allow constants in body
rnett Mar 5, 2021
768655f
Generalize allowing 0-input ops in the body
rnett Mar 5, 2021
8702dd3
Fixes, upstream and downstream functions
rnett Mar 9, 2021
09d30b5
Small fixes
rnett Mar 9, 2021
8a8440c
Back to ArrayDeque
rnett Mar 9, 2021
f5c52dd
Small fixes
rnett Mar 17, 2021
ff37a27
Change completeSubgraph signature, leave filtering to users
rnett Mar 17, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 11 additions & 0 deletions tensorflow-core/tensorflow-core-api/src/main/java/org/tensorflow/AbstractOperation.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,17 @@ public Output<?>[] outputList(int idx, int length) {

@Override
public <T extends TType> Output<T> output(int idx) {
if (getUnsafeNativeHandle(idx) != null && !getUnsafeNativeHandle(idx).isNull()) {
int numOutputs = this.numOutputs();
if (idx >= numOutputs) {
throw new IndexOutOfBoundsException(
"Can't get output with index " + idx + ", this op only has " + numOutputs + " outputs.");
}

if (idx < 0) {
throw new IndexOutOfBoundsException("Can't get output with index < 0.");
}
}
return new Output<>(this, idx);
}

Expand Down
Loading